Lines Matching defs:forward
95 def forward(self, x): member in TestUnconvertibleOps.setUp.EinsumModule
157 def forward(self, x): member in TestUnconvertibleOps.test_it_returns_empty_list_when_model_contains_supported_inplace_ops.SkipConnectionModule
188 def forward(self, x): member in TestUtilityFuns.test_is_in_onnx_export.MyModule
223 def forward(self, x, y, t): member in TestUtilityFuns.test_split_to_slice.SplitModule
244 def forward(self, x): member in TestUtilityFuns.test_constant_fold_transpose.TransposeModule
264 def forward(self, x): member in TestUtilityFuns.test_constant_fold_reduceL2.ReduceModule
282 def forward(self, x): member in TestUtilityFuns.test_constant_fold_reduceL1.NormModule
299 def forward(self, x): member in TestUtilityFuns.test_constant_fold_slice.NarrowModule
318 def forward(self, x): member in TestUtilityFuns.test_constant_fold_slice_index_exceeds_dim.SliceIndexExceedsDimModule
340 def forward(self, x): member in TestUtilityFuns.test_constant_fold_slice_negative_index.SliceNegativeIndexModule
363 def forward(self, x): member in TestUtilityFuns.test_constant_fold_gather.GatherModule
383 def forward(self, x): member in TestUtilityFuns.test_constant_fold_unsqueeze.UnsqueezeModule
406 def forward(self, x): member in TestUtilityFuns.test_constant_fold_unsqueeze_multi_axies.PReluModel
424 def forward(self, x): member in TestUtilityFuns.test_constant_fold_squeeze_without_axes.SqueezeModule
441 def forward(self, x): member in TestUtilityFuns.test_constant_fold_squeeze_with_axes.SqueezeAxesModule
459 def forward(self, x): member in TestUtilityFuns.test_constant_fold_concat.ConcatModule
499 def forward(self, input, initial_state): member in TestUtilityFuns.test_constant_fold_lstm.GruNet
530 def forward(self, A): member in TestUtilityFuns.test_constant_fold_transpose_matmul.MatMulNet
552 def forward(self, x): member in TestUtilityFuns.test_constant_fold_reshape.ReshapeModule
575 def forward(self, x): member in TestUtilityFuns.test_constant_fold_div.Module
598 def forward(self, x): member in TestUtilityFuns.test_constant_fold_mul.Module
621 def forward(self, x): member in TestUtilityFuns.test_constant_fold_add.Module
652 def forward(self, x): member in TestUtilityFuns.test_constant_fold_sub.Module
683 def forward(self, x): member in TestUtilityFuns.test_constant_fold_sqrt.Module
703 def forward(self, x): member in TestUtilityFuns.test_constant_fold_shape.ShapeModule
729 def forward(self, input): member in TestUtilityFuns.test_verbose.MyModule
767 def forward(self, x, y): member in TestUtilityFuns.test_sequence_dim.Module
807 def forward(self, x): member in TestUtilityFuns.test_export_mode.MyModule
843 def forward(self, x): member in TestUtilityFuns.test_export_does_not_fail_on_frozen_scripted_module.Inner
854 def forward(self, x): member in TestUtilityFuns.test_export_does_not_fail_on_frozen_scripted_module.Outer
873 def forward(self, x): member in TestUtilityFuns.test_local_function.N
887 def forward(self, x, y, z): member in TestUtilityFuns.test_local_function.M
985 def forward(self, x, y=None, z=None): member in TestUtilityFuns.test_local_function_overloads.NWithOverloads
998 def forward(self, x, y, z): member in TestUtilityFuns.test_local_function_overloads.M
1026 def forward(self, x): member in TestUtilityFuns.test_local_function_infer_scopes.M
1059 def forward(self, x): member in TestUtilityFuns.test_local_function_predefined_attributes.M
1122 def forward(self, x): member in TestUtilityFuns.test_local_function_subset_of_predefined_attributes.M
1146 def forward(self, x): member in TestUtilityFuns.test_node_scope.N
1160 def forward(self, x, y, z): member in TestUtilityFuns.test_node_scope.M
1208 def forward(self, x): member in TestUtilityFuns.test_scope_of_constants_when_combined_by_cse_pass.M
1220 def forward(self, x): member in TestUtilityFuns.test_scope_of_constants_when_combined_by_cse_pass.N
1257 def forward(self, x): member in TestUtilityFuns.test_scope_of_nodes_when_combined_by_cse_pass.M
1275 def forward(self, x): member in TestUtilityFuns.test_scope_of_nodes_when_combined_by_cse_pass.N
1311 def forward(self, x): member in TestUtilityFuns.test_aten_fallthrough.Module
1347 def forward(self, input, other): member in TestUtilityFuns.test_custom_op_fallthrough.FooModel
1413 def forward(self, x): member in TestUtilityFuns.test_custom_opsets_inverse.CustomInverse
1440 def forward(self, x): member in TestUtilityFuns.test_onnx_fallthrough.Module
1460 def forward(self, x): member in TestUtilityFuns.test_prim_fallthrough.PrimModule
1487 def forward(ctx, input): member in TestUtilityFuns.test_custom_layer_tuple.CustomFunction
1491 def forward(self, input): member in TestUtilityFuns.test_custom_layer_tuple.Custom
1506 def forward(ctx, input): member in TestUtilityFuns.test_autograd_onnx_fallthrough.CustomFunction
1518 def forward(self, input): member in TestUtilityFuns.test_autograd_onnx_fallthrough.Custom
1537 def forward(ctx, input): member in TestUtilityFuns.test_autograd_module_name.CustomFunction
1549 def forward(self, input): member in TestUtilityFuns.test_autograd_module_name.Custom
1578 def forward(self, x): member in TestUtilityFuns.test_unused_initializers.Model
1605 def forward(self, x): member in TestUtilityFuns.test_scripting_param.MyModule
1641 def forward(self, x): member in TestUtilityFuns.test_fuse_conv_bn.Fuse
1680 def forward(self, x, y): member in TestUtilityFuns.test_onnx_function_substitution_pass.MyModule
1711 def forward(self, x): member in TestUtilityFuns.test_onnx_value_name.MyModule
1746 def forward(self, x): member in TestUtilityFuns.test_onnx_node_naming.MainModule
1794 def forward(self, x): member in TestUtilityFuns._test_deduplicate_initializers.MyModule
1855 def forward(self, x, y): member in TestUtilityFuns.test_deduplicate_initializers_diff_devices.Model
1871 def forward(self, input0, input1): member in TestUtilityFuns.test_duplicated_output_node.DuplicatedOutputNet
1924 def forward(self, x): member in TestUtilityFuns.test_deduplicate_ignore_upsample_scale.Model
1952 def forward(self, x): member in TestUtilityFuns.test_bad_symbolic_registration.CatModel