Index _ | A | B | C | D | E | F | G | H | I | M | N | O | P | Q | R | S | T | U | V | W _ _retrieve() (in module experimental_experiment.torch_interpreter.onnx_export) A add_constant_node() (experimental_experiment.xbuilder.GraphBuilder method) add_domain() (experimental_experiment.xbuilder.GraphBuilder method) add_loss_output() (in module experimental_experiment.gradient.loss_helper) AddAdd (class in experimental_experiment.reference.evaluator) AddAddMulMulPattern (class in experimental_experiment.xoptim.patterns_exp.binary_operators) AddMul (class in experimental_experiment.reference.evaluator) AddMulBroadcastPattern (class in experimental_experiment.xoptim.patterns_exp.binary_operators) AddMulPattern (class in experimental_experiment.xoptim.patterns_exp.binary_operators) AddMulSharedInputBroadcastPattern (class in experimental_experiment.xoptim.patterns_exp.binary_operators) AddMulSharedInputPattern (class in experimental_experiment.xoptim.patterns_exp.binary_operators) AddMulTransposePattern (class in experimental_experiment.xoptim.patterns_exp.binary_operators) AddReductionScatterND (class in experimental_experiment.xoptim.patterns_fix.add_reduction_scatter_nd) AddSharedInput (class in experimental_experiment.reference.evaluator) apply() (experimental_experiment.xoptim.EasyPatternOptimization method) (experimental_experiment.xoptim.PatternOptimization method) apply_match() (experimental_experiment.xoptim.GraphBuilderPatternOptimization method) apply_pattern() (experimental_experiment.xoptim.EasyPatternOptimization method) assert_all_close() (in module experimental_experiment.torch_models.dump_helper) assertAlmostEqual() (experimental_experiment.ext_test_case.ExtTestCase method) aten__enter_autocast() (in module experimental_experiment.torch_interpreter._aten_functions) aten__exit_autocast() (in module experimental_experiment.torch_interpreter._aten_functions) aten__log_softmax() (in module experimental_experiment.torch_interpreter._aten_functions) aten__log_softmax_backward_data() (in module experimental_experiment.torch_interpreter._aten_functions) aten__prelu_kernel() (in module experimental_experiment.torch_interpreter._aten_functions) aten__prelu_kernel_backward() (in module experimental_experiment.torch_interpreter._aten_functions) aten__scaled_dot_product_flash_attention_for_cpu() (in module experimental_experiment.torch_interpreter._aten_functions) aten__set_grad_enabled() (in module experimental_experiment.torch_interpreter._aten_functions) aten__softmax() (in module experimental_experiment.torch_interpreter._aten_functions) aten__softmax_backward_data() (in module experimental_experiment.torch_interpreter._aten_functions) aten__to_copy() (in module experimental_experiment.torch_interpreter._aten_functions) aten__unsafe_index_put() (in module experimental_experiment.torch_interpreter._aten_functions) aten__unsafe_view() (in module experimental_experiment.torch_interpreter._aten_functions) aten_abs() (in module experimental_experiment.torch_interpreter._aten_functions) aten_acos() (in module experimental_experiment.torch_interpreter._aten_functions) aten_acosh() (in module experimental_experiment.torch_interpreter._aten_functions) aten_add() (in module experimental_experiment.torch_interpreter._aten_functions) aten_add_Scalar() (in module experimental_experiment.torch_interpreter._aten_functions) aten_add_Tensor() (in module experimental_experiment.torch_interpreter._aten_functions) aten_addmm() (in module experimental_experiment.torch_interpreter._aten_functions) aten_alias() (in module experimental_experiment.torch_interpreter._aten_functions) aten_all() (in module experimental_experiment.torch_interpreter._aten_functions) aten_amax() (in module experimental_experiment.torch_interpreter._aten_functions) aten_and() (in module experimental_experiment.torch_interpreter._aten_functions) aten_and_() (in module experimental_experiment.torch_interpreter._aten_functions) aten_arange() (in module experimental_experiment.torch_interpreter._aten_functions) aten_arange_start() (in module experimental_experiment.torch_interpreter._aten_functions) aten_arange_start_step() (in module experimental_experiment.torch_interpreter._aten_functions) aten_argmax() (in module experimental_experiment.torch_interpreter._aten_functions) aten_as_strided() (in module experimental_experiment.torch_interpreter._aten_functions) aten_asin() (in module experimental_experiment.torch_interpreter._aten_functions) aten_asinh() (in module experimental_experiment.torch_interpreter._aten_functions) aten_atan() (in module experimental_experiment.torch_interpreter._aten_functions) aten_atanh() (in module experimental_experiment.torch_interpreter._aten_functions) aten_bmm() (in module experimental_experiment.torch_interpreter._aten_functions) aten_cat() (in module experimental_experiment.torch_interpreter._aten_functions) aten_clone() (in module experimental_experiment.torch_interpreter._aten_functions) aten_conv2d() (in module experimental_experiment.torch_interpreter._aten_functions) aten_convolution() (in module experimental_experiment.torch_interpreter._aten_functions) aten_copy() (in module experimental_experiment.torch_interpreter._aten_functions) aten_copy_() (in module experimental_experiment.torch_interpreter._aten_functions) aten_cos() (in module experimental_experiment.torch_interpreter._aten_functions) aten_cosh() (in module experimental_experiment.torch_interpreter._aten_functions) aten_detach() (in module experimental_experiment.torch_interpreter._aten_functions) aten_div() (in module experimental_experiment.torch_interpreter._aten_functions) aten_div_Scalar() (in module experimental_experiment.torch_interpreter._aten_functions) aten_div_Tensor() (in module experimental_experiment.torch_interpreter._aten_functions) aten_dropout() (in module experimental_experiment.torch_interpreter._aten_functions) aten_elu() (in module experimental_experiment.torch_interpreter._aten_functions) aten_embedding() (in module experimental_experiment.torch_interpreter._aten_functions) aten_empty_like() (in module experimental_experiment.torch_interpreter._aten_functions) aten_empty_permuted() (in module experimental_experiment.torch_interpreter._aten_functions) aten_empty_strided() (in module experimental_experiment.torch_interpreter._aten_functions) aten_eq() (in module experimental_experiment.torch_interpreter._aten_functions) aten_eq_Scalar() (in module experimental_experiment.torch_interpreter._aten_functions) aten_eq_Tensor() (in module experimental_experiment.torch_interpreter._aten_functions) aten_exp() (in module experimental_experiment.torch_interpreter._aten_functions) aten_expand() (in module experimental_experiment.torch_interpreter._aten_functions) aten_fill_Scalar() (in module experimental_experiment.torch_interpreter._aten_functions) aten_flatten() (in module experimental_experiment.torch_interpreter._aten_functions) aten_full() (in module experimental_experiment.torch_interpreter._aten_functions) aten_full_like() (in module experimental_experiment.torch_interpreter._aten_functions) aten_FunctionCtx() (in module experimental_experiment.torch_interpreter._aten_functions) aten_ge() (in module experimental_experiment.torch_interpreter._aten_functions) aten_ge_Tensor() (in module experimental_experiment.torch_interpreter._aten_functions) aten_gt() (in module experimental_experiment.torch_interpreter._aten_functions) aten_gt_Tensor() (in module experimental_experiment.torch_interpreter._aten_functions) aten_index_put() (in module experimental_experiment.torch_interpreter._aten_functions) aten_index_select() (in module experimental_experiment.torch_interpreter._aten_functions) aten_index_Tensor() (in module experimental_experiment.torch_interpreter._aten_functions) aten_layer_norm() (in module experimental_experiment.torch_interpreter._aten_functions) aten_leaky_relu() (in module experimental_experiment.torch_interpreter._aten_functions) aten_leaky_relu_backward() (in module experimental_experiment.torch_interpreter._aten_functions) aten_linear() (in module experimental_experiment.torch_interpreter._aten_functions) aten_lt() (in module experimental_experiment.torch_interpreter._aten_functions) aten_lt_Tensor() (in module experimental_experiment.torch_interpreter._aten_functions) aten_masked_fill_Scalar() (in module experimental_experiment.torch_interpreter._aten_functions) aten_matmul() (in module experimental_experiment.torch_interpreter._aten_functions) aten_max_pool2d() (in module experimental_experiment.torch_interpreter._aten_functions) aten_max_pool2d_with_indices() (in module experimental_experiment.torch_interpreter._aten_functions) aten_mean_dim() (in module experimental_experiment.torch_interpreter._aten_functions) aten_meth_bool() (in module experimental_experiment.torch_interpreter._aten_methods) aten_meth_clone() (in module experimental_experiment.torch_interpreter._aten_methods) aten_meth_contiguous() (in module experimental_experiment.torch_interpreter._aten_methods) aten_meth_cos() (in module experimental_experiment.torch_interpreter._aten_methods) aten_meth_cpu() (in module experimental_experiment.torch_interpreter._aten_methods) aten_meth_eq() (in module experimental_experiment.torch_interpreter._aten_methods) aten_meth_expand() (in module experimental_experiment.torch_interpreter._aten_methods) aten_meth_float() (in module experimental_experiment.torch_interpreter._aten_methods) aten_meth_masked_fill() (in module experimental_experiment.torch_interpreter._aten_methods) aten_meth_masked_fill_() (in module experimental_experiment.torch_interpreter._aten_methods) aten_meth_mean() (in module experimental_experiment.torch_interpreter._aten_methods) aten_meth_pow() (in module experimental_experiment.torch_interpreter._aten_methods) aten_meth_repeat() (in module experimental_experiment.torch_interpreter._aten_methods) aten_meth_reshape() (in module experimental_experiment.torch_interpreter._aten_methods) aten_meth_sin() (in module experimental_experiment.torch_interpreter._aten_methods) aten_meth_size() (in module experimental_experiment.torch_interpreter._aten_methods) aten_meth_t() (in module experimental_experiment.torch_interpreter._aten_methods) aten_meth_to() (in module experimental_experiment.torch_interpreter._aten_methods) aten_meth_transpose() (in module experimental_experiment.torch_interpreter._aten_methods) aten_meth_unsqueeze() (in module experimental_experiment.torch_interpreter._aten_methods) aten_meth_view() (in module experimental_experiment.torch_interpreter._aten_methods) aten_mm() (in module experimental_experiment.torch_interpreter._aten_functions) aten_mul() (in module experimental_experiment.torch_interpreter._aten_functions) aten_mul_Scalar() (in module experimental_experiment.torch_interpreter._aten_functions) aten_mul_Tensor() (in module experimental_experiment.torch_interpreter._aten_functions) aten_native_dropout() (in module experimental_experiment.torch_interpreter._aten_functions) aten_native_layer_norm() (in module experimental_experiment.torch_interpreter._aten_functions) aten_ne() (in module experimental_experiment.torch_interpreter._aten_functions) aten_ne_Tensor() (in module experimental_experiment.torch_interpreter._aten_functions) aten_neg() (in module experimental_experiment.torch_interpreter._aten_functions) aten_new_zeros() (in module experimental_experiment.torch_interpreter._aten_functions) aten_not() (in module experimental_experiment.torch_interpreter._aten_functions) aten_not_() (in module experimental_experiment.torch_interpreter._aten_functions) aten_ones() (in module experimental_experiment.torch_interpreter._aten_functions) aten_ones_like() (in module experimental_experiment.torch_interpreter._aten_functions) aten_permute() (in module experimental_experiment.torch_interpreter._aten_functions) aten_pow_Tensor_Scalar() (in module experimental_experiment.torch_interpreter._aten_functions) aten_pow_Tensor_Tensor() (in module experimental_experiment.torch_interpreter._aten_functions) aten_relu() (in module experimental_experiment.torch_interpreter._aten_functions) aten_repeat() (in module experimental_experiment.torch_interpreter._aten_functions) aten_round() (in module experimental_experiment.torch_interpreter._aten_functions) aten_rrelu_with_noise_backward() (in module experimental_experiment.torch_interpreter._aten_functions) aten_rsqrt() (in module experimental_experiment.torch_interpreter._aten_functions) aten_rsub() (in module experimental_experiment.torch_interpreter._aten_functions) aten_rsub_Scalar() (in module experimental_experiment.torch_interpreter._aten_functions) aten_scaled_dot_product_attention() (in module experimental_experiment.torch_interpreter._aten_functions) aten_select_int() (in module experimental_experiment.torch_interpreter._aten_functions) aten_setitem() (in module experimental_experiment.torch_interpreter._aten_functions) aten_sigmoid() (in module experimental_experiment.torch_interpreter._aten_functions) aten_sigmoid_backward() (in module experimental_experiment.torch_interpreter._aten_functions) aten_silu() (in module experimental_experiment.torch_interpreter._aten_functions) aten_sin() (in module experimental_experiment.torch_interpreter._aten_functions) aten_sinh() (in module experimental_experiment.torch_interpreter._aten_functions) aten_slice_backward() (in module experimental_experiment.torch_interpreter._aten_functions) aten_slice_scatter() (in module experimental_experiment.torch_interpreter._aten_functions) aten_slice_Tensor() (in module experimental_experiment.torch_interpreter._aten_functions) aten_softmax() (in module experimental_experiment.torch_interpreter._aten_functions) aten_softmax_int() (in module experimental_experiment.torch_interpreter._aten_functions) aten_split_with_sizes() (in module experimental_experiment.torch_interpreter._aten_functions) aten_sqrt() (in module experimental_experiment.torch_interpreter._aten_functions) aten_squeeze() (in module experimental_experiment.torch_interpreter._aten_functions) aten_squeeze_dim() (in module experimental_experiment.torch_interpreter._aten_functions) aten_sub() (in module experimental_experiment.torch_interpreter._aten_functions) aten_sub_Tensor() (in module experimental_experiment.torch_interpreter._aten_functions) aten_sum() (in module experimental_experiment.torch_interpreter._aten_functions) aten_sum_dim_IntList() (in module experimental_experiment.torch_interpreter._aten_functions) aten_t() (in module experimental_experiment.torch_interpreter._aten_functions) aten_tan() (in module experimental_experiment.torch_interpreter._aten_functions) aten_tanh() (in module experimental_experiment.torch_interpreter._aten_functions) aten_tanh_backward() (in module experimental_experiment.torch_interpreter._aten_functions) aten_tensor() (in module experimental_experiment.torch_interpreter._aten_functions) aten_threshold_backward() (in module experimental_experiment.torch_interpreter._aten_functions) aten_transpose() (in module experimental_experiment.torch_interpreter._aten_functions) aten_transpose_int() (in module experimental_experiment.torch_interpreter._aten_functions) aten_tril() (in module experimental_experiment.torch_interpreter._aten_functions) aten_triu() (in module experimental_experiment.torch_interpreter._aten_functions) aten_truediv() (in module experimental_experiment.torch_interpreter._aten_functions) aten_unsqueeze() (in module experimental_experiment.torch_interpreter._aten_functions) aten_view() (in module experimental_experiment.torch_interpreter._aten_functions) aten_zeros() (in module experimental_experiment.torch_interpreter._aten_functions) B build_matching_inputs() (in module experimental_experiment.torch_models.dump_helper) C call_function() (experimental_experiment.torch_interpreter.interpreter.DynamoInterpreter method) call_method() (experimental_experiment.torch_interpreter.interpreter.DynamoInterpreter method) call_module() (experimental_experiment.torch_interpreter.interpreter.DynamoInterpreter method) capture() (experimental_experiment.ext_test_case.ExtTestCase method) CastCastBinaryPattern (class in experimental_experiment.xoptim.patterns) CastLike_15 (class in experimental_experiment.reference.evaluator) CastLike_19 (class in experimental_experiment.reference.evaluator) CastOpCastPattern (class in experimental_experiment.xoptim.patterns) CastPattern (class in experimental_experiment.xoptim.patterns) check_model_ort() (in module experimental_experiment.torch_test_helper) choose_consistent_domain_opset() (in module experimental_experiment.xbuilder._onnx_helper) compatible_dimensions() (in module experimental_experiment.xbuilder.shape_helper) compatible_opsets() (in module experimental_experiment.xbuilder._onnx_helper) compatible_shapes() (in module experimental_experiment.xbuilder.shape_helper) ComputationCastOpCastPattern (class in experimental_experiment.xoptim.patterns) Concat (class in experimental_experiment.reference.evaluator) constant_folding() (experimental_experiment.xbuilder.GraphBuilder method) ConstantOfShape (class in experimental_experiment.reference.evaluator) ConstantOfShapeScatterNDPattern (class in experimental_experiment.xoptim.patterns_exp.constant_of_shape_scatter_nd) D debug_string() (experimental_experiment.xoptim.MatchResult method) DerivativeOptions (class in experimental_experiment.gradient.grad_helper) Dispatcher (class in experimental_experiment.torch_interpreter) display_pattern() (experimental_experiment.xoptim.EasyPatternOptimization method) DivByMulScalarPattern (class in experimental_experiment.xoptim.patterns) do_not_remove() (experimental_experiment.xbuilder.GraphBuilder method) (experimental_experiment.xoptim.GraphBuilderPatternOptimization method) dump_onnx() (experimental_experiment.ext_test_case.ExtTestCase method) (in module experimental_experiment.torch_models.dump_helper) DynamoInterpreter (class in experimental_experiment.torch_interpreter.interpreter) dynger_backend() (in module experimental_experiment.torch_dynamo) E EasyPatternOptimization (class in experimental_experiment.xoptim) elem_size() (experimental_experiment.xbuilder.GraphBuilder method) element_wise_binary_op_types() (in module experimental_experiment.xbuilder._onnx_helper) element_wise_op_cmp_types() (in module experimental_experiment.xbuilder._onnx_helper) empty_copy() (experimental_experiment.xbuilder.GraphBuilder method) enumerate_matches() (experimental_experiment.xoptim.PatternOptimization method) ExpandBroadcastPattern (class in experimental_experiment.xoptim.patterns) ExpandPattern (class in experimental_experiment.xoptim.patterns) ExpandSwapPattern (class in experimental_experiment.xoptim.patterns) experimental_experiment.ext_test_case module experimental_experiment.torch_interpreter._aten_functions module experimental_experiment.torch_interpreter._aten_methods module experimental_experiment.torch_interpreter._prims_functions module experimental_experiment.torch_test_helper module export_to_onnx() (in module experimental_experiment.torch_test_helper) Expression (class in experimental_experiment.xbuilder.expression_dimension) ExtendedReferenceEvaluator (class in experimental_experiment.reference) ExtTestCase (class in experimental_experiment.ext_test_case) F fallback() (experimental_experiment.torch_interpreter.Dispatcher method) (experimental_experiment.torch_interpreter.ForceDispatcher method) (experimental_experiment.torch_interpreter.oxs_dispatcher.OxsDebugDispatcher method) (experimental_experiment.torch_interpreter.oxs_dispatcher.OxsDispatcher method) filter_decomposition_table() (in module experimental_experiment.torch_dynamo) find_function() (experimental_experiment.torch_interpreter.Dispatcher method) find_method() (experimental_experiment.torch_interpreter.Dispatcher method) ForceDispatcher (class in experimental_experiment.torch_interpreter) from_array_extended() (in module experimental_experiment.reference) FusedMatMul (class in experimental_experiment.reference.evaluator) FusedMatMulPattern (class in experimental_experiment.xoptim.patterns_ort.fused_matmul) FusedMatMulTransposePattern (class in experimental_experiment.xoptim.patterns_ort.fused_matmul) FusedMatMulx2Pattern (class in experimental_experiment.xoptim.patterns_ort.fused_matmul) G GatherGrad (class in experimental_experiment.reference.evaluator) GatherGradPattern (class in experimental_experiment.xoptim.patterns_ort.gather_grad) get_attr() (experimental_experiment.torch_interpreter.interpreter.DynamoInterpreter method) get_attribute() (experimental_experiment.xbuilder.GraphBuilder method) (experimental_experiment.xoptim.GraphBuilderPatternOptimization method) get_attributes_with_default() (experimental_experiment.xbuilder.GraphBuilder method) (experimental_experiment.xoptim.GraphBuilderPatternOptimization method) get_axis() (experimental_experiment.xoptim.GraphBuilderPatternOptimization method) get_computed_constant() (experimental_experiment.xoptim.GraphBuilderPatternOptimization method) get_constant() (experimental_experiment.xbuilder.GraphBuilder method) get_constant_or_attribute() (experimental_experiment.xbuilder.GraphBuilder method) (experimental_experiment.xoptim.GraphBuilderPatternOptimization method) get_constant_scalar() (experimental_experiment.xoptim.GraphBuilderPatternOptimization method) get_debug_msg() (experimental_experiment.xbuilder.GraphBuilder method) get_decomposition_table() (in module experimental_experiment.torch_dynamo) get_decomposition_table_dynamo() (in module experimental_experiment.torch_dynamo) get_default_patterns() (in module experimental_experiment.xoptim.patterns) get_dump_file() (experimental_experiment.ext_test_case.ExtTestCase method) get_figure() (in module experimental_experiment.ext_test_case) get_fix_patterns() (in module experimental_experiment.xoptim.patterns_fix) get_is_dimension() (experimental_experiment.xbuilder.GraphBuilder method) get_llama_attention() (in module experimental_experiment.torch_models.llama_helper) get_llama_decoder() (in module experimental_experiment.torch_models.llama_helper) get_llama_model() (in module experimental_experiment.torch_models.llama_helper) get_mistral_model() (in module experimental_experiment.torch_models.mistral_helper) get_onnxruntime_patterns() (in module experimental_experiment.xoptim.patterns_ort) get_opset() (experimental_experiment.xbuilder.GraphBuilder method) get_pattern() (in module experimental_experiment.xoptim) get_pattern_list() (in module experimental_experiment.xoptim) get_phi_model() (in module experimental_experiment.torch_models.phi_helper) get_rank() (experimental_experiment.xbuilder.GraphBuilder method) (experimental_experiment.xoptim.GraphBuilderPatternOptimization method) get_sequence() (experimental_experiment.xbuilder.GraphBuilder method) get_shape() (experimental_experiment.xbuilder.GraphBuilder method) (experimental_experiment.xoptim.GraphBuilderPatternOptimization method) get_train_initializer() (in module experimental_experiment.gradient.loss_helper) get_type() (experimental_experiment.xbuilder.GraphBuilder method) (experimental_experiment.xoptim.GraphBuilderPatternOptimization method) getitem() (experimental_experiment.torch_interpreter.interpreter.DynamoInterpreter method) GraphBuilder (class in experimental_experiment.xbuilder) GraphBuilderPatternOptimization (class in experimental_experiment.xoptim) H has_cuda() (in module experimental_experiment.ext_test_case) has_dynamic_object() (experimental_experiment.xbuilder.GraphBuilder method) has_name() (experimental_experiment.xbuilder.GraphBuilder method) has_processor() (experimental_experiment.xoptim.GraphBuilderPatternOptimization method) has_rank() (experimental_experiment.xbuilder.GraphBuilder method) (experimental_experiment.xoptim.GraphBuilderPatternOptimization method) has_shape() (experimental_experiment.xbuilder.GraphBuilder method) (experimental_experiment.xoptim.GraphBuilderPatternOptimization method) has_type() (experimental_experiment.xbuilder.GraphBuilder method) (experimental_experiment.xoptim.GraphBuilderPatternOptimization method) hide_stdout() (in module experimental_experiment.ext_test_case) I IdentityPattern (class in experimental_experiment.xoptim.patterns) ignore_warnings() (in module experimental_experiment.ext_test_case) inline_model_proto() (in module experimental_experiment.convert.convert_helper) input_names (experimental_experiment.xoptim.GraphBuilderPatternOptimization property) inputs (experimental_experiment.xoptim.GraphBuilderPatternOptimization property) inputs_from_onnx_model() (in module experimental_experiment.torch_models.dump_helper) insert_and_remove_nodes() (experimental_experiment.xbuilder.GraphBuilder method) io_names() (experimental_experiment.xbuilder.GraphBuilder method) is_azure() (in module experimental_experiment.ext_test_case) is_constant() (experimental_experiment.xbuilder.GraphBuilder method) (experimental_experiment.xoptim.GraphBuilderPatternOptimization method) is_constant_or_attribute() (experimental_experiment.xbuilder.GraphBuilder method) is_constant_scalar() (experimental_experiment.xoptim.GraphBuilderPatternOptimization method) is_exact_same_constant() (experimental_experiment.xbuilder.GraphBuilder method) is_output() (experimental_experiment.xoptim.GraphBuilderPatternOptimization method) is_sequence() (experimental_experiment.xbuilder.GraphBuilder method) is_used() (experimental_experiment.xoptim.GraphBuilderPatternOptimization method) is_used_by_subgraph() (experimental_experiment.xoptim.GraphBuilderPatternOptimization method) is_used_more_than_once() (experimental_experiment.xoptim.GraphBuilderPatternOptimization method) is_used_only_by() (experimental_experiment.xoptim.GraphBuilderPatternOptimization method) iter_nodes() (experimental_experiment.xoptim.GraphBuilderPatternOptimization method) M main_opset (experimental_experiment.xbuilder.GraphBuilder property) (experimental_experiment.xoptim.GraphBuilderPatternOptimization property) make_dynamic_object() (experimental_experiment.xbuilder.GraphBuilder method) make_initializer() (experimental_experiment.xbuilder.GraphBuilder method) make_key() (experimental_experiment.xbuilder.GraphBuilder method) make_node() (experimental_experiment.torch_interpreter.oxs_opset.OxsOpset method) (experimental_experiment.xbuilder.GraphBuilder method) (experimental_experiment.xoptim.GraphBuilderPatternOptimization method) make_node_check_opset() (experimental_experiment.xoptim.GraphBuilderPatternOptimization method) make_nodes() (experimental_experiment.xbuilder.GraphBuilder method) make_shape_from_results() (experimental_experiment.xbuilder.GraphBuilder method) make_tensor_input() (experimental_experiment.xbuilder.GraphBuilder method) make_tensor_output() (experimental_experiment.xbuilder.GraphBuilder method) MaskedScatterNDOfShape (class in experimental_experiment.reference.evaluator) MaskedShapeScatterNDPattern (class in experimental_experiment.xoptim.patterns_exp.constant_of_shape_scatter_nd) match() (experimental_experiment.xoptim.EasyPatternOptimization method) (experimental_experiment.xoptim.PatternOptimization method) match_pattern() (experimental_experiment.xoptim.EasyPatternOptimization method) MatchResult (class in experimental_experiment.xoptim) MatMulReshape2Of3Pattern (class in experimental_experiment.xoptim.patterns) measure_time() (in module experimental_experiment.ext_test_case) MemcpyFromHost (class in experimental_experiment.reference.evaluator) MemcpyToHost (class in experimental_experiment.reference.evaluator) module experimental_experiment.ext_test_case experimental_experiment.torch_interpreter._aten_functions experimental_experiment.torch_interpreter._aten_methods experimental_experiment.torch_interpreter._prims_functions experimental_experiment.torch_test_helper MulAdd (class in experimental_experiment.reference.evaluator) MulMul (class in experimental_experiment.reference.evaluator) MulMulMulScalarPattern (class in experimental_experiment.xoptim.patterns) MulSharedInput (class in experimental_experiment.reference.evaluator) MulSigmoid (class in experimental_experiment.reference.evaluator) MulSigmoidPattern (class in experimental_experiment.xoptim.patterns_exp.binary_operators) MulSub (class in experimental_experiment.reference.evaluator) N NegXplus1 (class in experimental_experiment.reference.evaluator) NegXplus1Pattern (class in experimental_experiment.xoptim.patterns_exp.binary_operators) next_node() (experimental_experiment.xoptim.GraphBuilderPatternOptimization method) next_nodes() (experimental_experiment.xoptim.GraphBuilderPatternOptimization method) node_before() (experimental_experiment.xoptim.GraphBuilderPatternOptimization method) nodes (experimental_experiment.xoptim.GraphBuilderPatternOptimization property) none() (experimental_experiment.xoptim.PatternOptimization method) O onnx_custom_backend() (in module experimental_experiment.torch_dynamo) onnx_debug_backend() (in module experimental_experiment.torch_dynamo) onnx_derivative() (in module experimental_experiment.gradient.grad_helper) onnx_lighten() (in module experimental_experiment.onnx_tools) onnx_unlighten() (in module experimental_experiment.onnx_tools) Opset (class in experimental_experiment.xbuilder.graph_builder) opsets (experimental_experiment.xoptim.GraphBuilderPatternOptimization property) OptimizationOptions (class in experimental_experiment.xbuilder) optimize() (experimental_experiment.xbuilder.GraphBuilder method) (experimental_experiment.xoptim.GraphBuilderPatternOptimization method) optimize_model_proto() (in module experimental_experiment.convert.convert_helper) optimize_with_patterns() (experimental_experiment.xbuilder.GraphBuilder method) ort_optimize() (in module experimental_experiment.convert.convert_helper) OrtBackend() (in module experimental_experiment.torch_dynamo.fast_backend) OrtEval (class in experimental_experiment.reference) output() (experimental_experiment.torch_interpreter.interpreter.DynamoInterpreter method) output_names (experimental_experiment.xoptim.GraphBuilderPatternOptimization property) outputs (experimental_experiment.xoptim.GraphBuilderPatternOptimization property) OxsDebugDispatcher (class in experimental_experiment.torch_interpreter.oxs_dispatcher) OxsDispatcher (class in experimental_experiment.torch_interpreter.oxs_dispatcher) OxsOpset (class in experimental_experiment.torch_interpreter.oxs_opset) P parse_dimension_expression() (experimental_experiment.xbuilder.GraphBuilder method) parse_expression() (in module experimental_experiment.xbuilder.expression_dimension) PatternOptimization (class in experimental_experiment.xoptim) penalty_loss_onnx() (in module experimental_experiment.gradient.loss_helper) placeholder() (experimental_experiment.torch_interpreter.interpreter.DynamoInterpreter method) prims_add() (in module experimental_experiment.torch_interpreter._prims_functions) prims_amax() (in module experimental_experiment.torch_interpreter._prims_functions) prims_broadcast_in_dim() (in module experimental_experiment.torch_interpreter._prims_functions) prims_cat() (in module experimental_experiment.torch_interpreter._prims_functions) prims_clone() (in module experimental_experiment.torch_interpreter._prims_functions) prims_collapse_view() (in module experimental_experiment.torch_interpreter._prims_functions) prims_convert_element_type() (in module experimental_experiment.torch_interpreter._prims_functions) prims_cos() (in module experimental_experiment.torch_interpreter._prims_functions) prims_div() (in module experimental_experiment.torch_interpreter._prims_functions) prims_empty_strided() (in module experimental_experiment.torch_interpreter._prims_functions) prims_eq() (in module experimental_experiment.torch_interpreter._prims_functions) prims_exp() (in module experimental_experiment.torch_interpreter._prims_functions) prims_ge() (in module experimental_experiment.torch_interpreter._prims_functions) prims_gt() (in module experimental_experiment.torch_interpreter._prims_functions) prims_iota() (in module experimental_experiment.torch_interpreter._prims_functions) prims_lt() (in module experimental_experiment.torch_interpreter._prims_functions) prims_mul() (in module experimental_experiment.torch_interpreter._prims_functions) prims_neg() (in module experimental_experiment.torch_interpreter._prims_functions) prims_pow() (in module experimental_experiment.torch_interpreter._prims_functions) prims_rsqrt() (in module experimental_experiment.torch_interpreter._prims_functions) prims_sin() (in module experimental_experiment.torch_interpreter._prims_functions) prims_split_dim() (in module experimental_experiment.torch_interpreter._prims_functions) prims_sub() (in module experimental_experiment.torch_interpreter._prims_functions) prims_sum() (in module experimental_experiment.torch_interpreter._prims_functions) prims_transpose() (in module experimental_experiment.torch_interpreter._prims_functions) prims_view_of() (in module experimental_experiment.torch_interpreter._prims_functions) prims_where() (in module experimental_experiment.torch_interpreter._prims_functions) Q QuickGelu (class in experimental_experiment.reference.evaluator) R rank() (experimental_experiment.xbuilder.GraphBuilder method) ReduceReshapePattern (class in experimental_experiment.xoptim.patterns) ReduceSumNormalizePattern (class in experimental_experiment.xoptim.patterns) remove_identity_nodes() (experimental_experiment.xbuilder.GraphBuilder method) remove_unused() (experimental_experiment.xbuilder.GraphBuilder method) reorder_functions_in_proto() (in module experimental_experiment.torch_models.dump_helper) ReplaceZero (class in experimental_experiment.reference.evaluator) ReplaceZeroPattern (class in experimental_experiment.xoptim.patterns_exp.where_replace) requires_onnx() (in module experimental_experiment.ext_test_case) requires_onnxruntime() (in module experimental_experiment.ext_test_case) requires_onnxruntime_training() (in module experimental_experiment.ext_test_case) requires_onnxscript() (in module experimental_experiment.ext_test_case) requires_torch() (in module experimental_experiment.ext_test_case) requires_transformers() (in module experimental_experiment.ext_test_case) requires_zoo() (in module experimental_experiment.ext_test_case) Reshape2Of3Pattern (class in experimental_experiment.xoptim.patterns) ReshapeMatMulReshapePattern (class in experimental_experiment.xoptim.patterns) ReshapeReshapeBinaryPattern (class in experimental_experiment.xoptim.patterns) ReshapeReshapePattern (class in experimental_experiment.xoptim.patterns) Rotary (class in experimental_experiment.reference.evaluator) RotaryConcatPartPattern (class in experimental_experiment.xoptim.patterns) run() (experimental_experiment.reference.ExtendedReferenceEvaluator method) (experimental_experiment.reference.OrtEval method) run_dlpack() (experimental_experiment.reference.OrtEval method) run_node() (experimental_experiment.torch_interpreter.interpreter.DynamoInterpreter method) S SameChildrenPattern (class in experimental_experiment.xoptim.patterns) ScatterElements (class in experimental_experiment.reference.evaluator) ScatterNDOfShape (class in experimental_experiment.reference.evaluator) select_outputs() (experimental_experiment.xbuilder.GraphBuilder method) set_name() (experimental_experiment.xbuilder.GraphBuilder method) set_rank() (experimental_experiment.xbuilder.GraphBuilder method) set_sequence() (experimental_experiment.xbuilder.GraphBuilder method) set_shape() (experimental_experiment.xbuilder.GraphBuilder method) set_type() (experimental_experiment.xbuilder.GraphBuilder method) set_value_shape() (experimental_experiment.xbuilder.GraphBuilder method) SimpleRotaryPattern (class in experimental_experiment.xoptim.patterns_exp.simple_rotary) SimplifiedLayerNormalization (class in experimental_experiment.reference.evaluator) SimplifiedLayerNormalizationPattern (class in experimental_experiment.xoptim.patterns_ort.simplified_layer_normalization) skipif_ci_apple() (in module experimental_experiment.ext_test_case) skipif_ci_windows() (in module experimental_experiment.ext_test_case) skipif_not_onnxrt() (in module experimental_experiment.ext_test_case) skipif_transformers() (in module experimental_experiment.ext_test_case) Slice_1 (class in experimental_experiment.reference.evaluator) Slice_10 (class in experimental_experiment.reference.evaluator) SlicesSplitPattern (class in experimental_experiment.xoptim.patterns) SoftmaxGradPattern (class in experimental_experiment.xoptim.patterns_ort.activation_grad) statistics_on_file() (in module experimental_experiment.ext_test_case) statistics_on_folder() (in module experimental_experiment.ext_test_case) Sub1MulPattern (class in experimental_experiment.xoptim.patterns) submodules (experimental_experiment.torch_interpreter.oxs_dispatcher.OxsDispatcher property) SubMul (class in experimental_experiment.reference.evaluator) SubMulBroadcastPattern (class in experimental_experiment.xoptim.patterns_exp.binary_operators) SubMulPattern (class in experimental_experiment.xoptim.patterns_exp.binary_operators) SwitchOrderBinaryPattern (class in experimental_experiment.xoptim.patterns) T tearDownClass() (experimental_experiment.ext_test_case.ExtTestCase class method) to_onnx() (experimental_experiment.xbuilder.GraphBuilder method) (in module experimental_experiment.torch_interpreter) TorchModelContainer (class in experimental_experiment.xbuilder.model_container) Transpose2DCastFP16 (class in experimental_experiment.reference.evaluator) Transpose2DCastFP32 (class in experimental_experiment.reference.evaluator) TransposeCastPattern (class in experimental_experiment.xoptim.patterns_exp.unary_operators) TransposeMatMulPattern (class in experimental_experiment.xoptim.patterns) TransposeReshapeMatMulPattern (class in experimental_experiment.xoptim.patterns) TransposeTransposePattern (class in experimental_experiment.xoptim.patterns) TriMatrix (class in experimental_experiment.reference.evaluator) TriMatrixPattern (class in experimental_experiment.xoptim.patterns_exp.constants) try_infer_shape() (experimental_experiment.xoptim.GraphBuilderPatternOptimization method) try_infer_type() (experimental_experiment.xoptim.GraphBuilderPatternOptimization method) tryCall() (experimental_experiment.ext_test_case.ExtTestCase method) U unary_like_op_types() (in module experimental_experiment.xbuilder._onnx_helper) unique_name() (experimental_experiment.xoptim.GraphBuilderPatternOptimization method) unit_test_going() (in module experimental_experiment.ext_test_case) UnsqueezeEqualPattern (class in experimental_experiment.xoptim.patterns) UnsqueezeUnsqueezePattern (class in experimental_experiment.xoptim.patterns) V value_as_shape() (experimental_experiment.xbuilder.GraphBuilder method) verify_dynamic_shape() (experimental_experiment.xbuilder.GraphBuilder method) W with_path_append() (in module experimental_experiment.ext_test_case)