| /external/pytorch/test/mobile/model_test/ |
| D | nn_ops.py | 34 def forward(self): member in NNConvolutionModule
82 def forward(self): member in NNPoolingModule
123 def forward(self): member in NNPaddingModule
164 def forward(self): member in NNNormalizationModule
212 def forward(self): member in NNActivationModule
234 def forward(self): member in NNRecurrentModule
264 def forward(self): member in NNTransformerModule
285 def forward(self): member in NNLinearModule
294 def forward(self): member in NNDropoutModule
308 def forward(self): member in NNSparseModule
[all …]
|
| /external/executorch/backends/qualcomm/tests/ |
| D | models.py | 15 def forward(self, x, y): member in Add
23 def forward(self, x): member in AddConstantFloat
31 def forward(self, x): member in AddConstantLong
40 def forward(self, y): member in Arange
54 def forward(self, x): member in AvgPoolModule
64 def forward(self, x): member in BatchNorm
72 def forward(self, x, y): member in Bmm
80 def forward(self, x): member in Cast
88 def forward(self, x, y): member in Cat2
96 def forward(self, x, y): member in Cat3
[all …]
|
| /external/pytorch/test/onnx/ |
| D | test_pytorch_onnx_onnxruntime.py | 174 def forward(self, x): member in TestONNXRuntime.test_fuse_conv_bn1d.Fuse
191 def forward(self, x): member in TestONNXRuntime.test_fuse_conv_bn2d.Fuse
208 def forward(self, x): member in TestONNXRuntime.test_fuse_conv_bn3d.Fuse
230 def forward(self, x): member in TestONNXRuntime.test_fuse_conv_in_block.Fuse
279 def forward(self, input): member in TestONNXRuntime.test_conv_tbc.ConvTBC
297 def forward(self, x): member in TestONNXRuntime.test_reshape_constant_fold.Reshape
414 def forward(self, input): member in TestONNXRuntime.test_index_1d.MyModel
422 def forward(self, input): member in TestONNXRuntime.test_index_2d_1dimslice.MyModel
430 def forward(self, input): member in TestONNXRuntime.test_index_2d_sliceint.MyModel
438 def forward(self, input): member in TestONNXRuntime.test_index_2d_neg_slice.MyModel
[all …]
|
| D | test_autograd_funs.py | 21 def forward(ctx, i): member in TestAutogradFuns.test_single_output.SingleOut
33 def forward(self, input): member in TestAutogradFuns.test_single_output.Caller
44 def forward(ctx, i): member in TestAutogradFuns.test_multi_output.MultiOut
56 def forward(self, input): member in TestAutogradFuns.test_multi_output.Caller
66 def forward(ctx, input): member in TestAutogradFuns.test_partial_output.PartialOut
72 def forward(self, input): member in TestAutogradFuns.test_partial_output.Caller
82 def forward(ctx, i): member in TestAutogradFuns.test_nested_autograd.Child
95 def forward(ctx, i): member in TestAutogradFuns.test_nested_autograd.Parent
107 def forward(self, input): member in TestAutogradFuns.test_nested_autograd.Caller
118 def forward(ctx, x): member in TestAutogradFuns.test_aten_unsupported.Erf
[all …]
|
| D | test_utility_funs.py | 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
[all …]
|
| D | test_pytorch_onnx_no_runtime.py | 81 def forward(self, x): member in TestONNXExport.test_fuse_addmm.AddmmModel
96 def forward(self, x): member in TestONNXExport.test_onnx_transpose_incomplete_tensor_type.Foo
104 def forward(self, x): member in TestONNXExport.test_onnx_transpose_incomplete_tensor_type.TraceMe
123 def forward(self, x): member in TestONNXExport.test_onnx_export_script_module.ModuleToExport
137 def forward(self, x): member in TestONNXExport.test_onnx_export_func_with_warnings.WarningTest
148 def forward(self, x): member in TestONNXExport.test_onnx_export_script_python_fail.PythonModule
157 def forward(self, x): member in TestONNXExport.test_onnx_export_script_python_fail.ModuleToExport
168 def forward(self, x): member in TestONNXExport.test_onnx_export_script_inline_trace.ModuleToInline
177 def forward(self, x): member in TestONNXExport.test_onnx_export_script_inline_trace.ModuleToExport
187 def forward(self, x): member in TestONNXExport.test_onnx_export_script_inline_script.ModuleToInline
[all …]
|
| /external/pytorch/test/export/ |
| D | test_export.py | 203 def forward(self, x): member in TestDynamismExpression.test_export_inline_constraints.Module
223 def forward(self, x): member in TestDynamismExpression.test_export_constraints_error_not_in_range.InvalidInputConflictWithInputConstraints
237 def forward(self, *args): member in TestDynamismExpression.test_export_slice_maxsize.Slice
250 def forward(self, x): member in TestDynamismExpression.test_export_constraints_error.ConflictingConstraints
269 def forward(self, x: torch.Tensor): member in TestDynamismExpression.test_export_assume_static_by_default.Module
343 def forward(self, x, y): member in TestExport.test_basic.Module
352 def forward(self, x, y): member in TestExport.test_no_tensor_computation.Module
370 def forward(self, x, y): member in TestExport.test_no_tensor_computation_2.Module
388 def forward(self, x, y): member in TestExport.test_no_tensor_computation_3.Module
406 def forward(self, x, y): member in TestExport.test_no_tensor_computation_4.Module
[all …]
|
| D | test_converter.py | 140 def forward(self, x, y): member in TestConverter.test_ts2ep_converter_basic.MSingle
144 def forward(self, x, y): member in TestConverter.test_ts2ep_converter_basic.MMulti
156 def forward(self, x: torch.Tensor, y: torch.Tensor): member in TestConverter.test_ts2ep_converter_container_output.MOutputList
163 def forward(self, x: torch.Tensor, y: torch.Tensor): member in TestConverter.test_ts2ep_converter_container_output.MOutputTuple
170 def forward(self, x: torch.Tensor, y: torch.Tensor): member in TestConverter.test_ts2ep_converter_container_output.MOutputDict
184 def forward(self, x): member in TestConverter.test_aten_dim.Module
193 def forward(self, x: torch.Tensor): member in TestConverter.test_aten_len.Module
202 def forward(self, x: List[int]): member in TestConverter.test_aten_len.Module
211 def forward(self, x: Dict[int, str]): member in TestConverter.test_aten_len.Module
220 def forward(self, x: Dict[bool, str]): member in TestConverter.test_aten_len.Module
[all …]
|
| D | test_unflatten.py | 58 def forward(self, x): member in TestUnflatten.test_unflatten_nested.NestedChild
69 def forward(self, x): member in TestUnflatten.test_unflatten_nested.Child1
78 def forward(self, x): member in TestUnflatten.test_unflatten_nested.Child2
90 def forward(self, x): member in TestUnflatten.test_unflatten_nested.MyModule
120 def forward(self, x): member in TestUnflatten.test_unflatten_buffer_mutation.Child
132 def forward(self, x): member in TestUnflatten.test_unflatten_buffer_mutation.MyModule
156 def forward(self, x): member in TestUnflatten.test_unflatten_nested_access.Child
167 def forward(self, x): member in TestUnflatten.test_unflatten_nested_access.MyModule
191 def forward(self, x): member in TestUnflatten.test_unflatten_shared_submodule.Shared
211 def forward(self, zx, y): member in TestUnflatten.test_unflatten_preserve_signature.NestedChild
[all …]
|
| D | test_experimental.py | 25 def forward(self, x): member in TestExperiment.test_with_buffer_as_submodule.B
38 def forward(self, x): member in TestExperiment.test_with_buffer_as_submodule.M
92 def forward(self, x): member in TestExperiment.test_mark_strict_with_container_type.B
101 def forward(self, x): member in TestExperiment.test_mark_strict_with_container_type.M
112 def forward(self, x): member in TestExperiment.test_torchscript_module_export.M
127 def forward(self, x): member in TestExperiment.test_torchscript_module_export_single_input.M
174 def forward(self, x: Tuple[torch.Tensor]): member in TestExperiment.test_torchscript_module_export_various_inputs_with_annotated_input_names.MTuple
181 def forward(self, x: List[torch.Tensor]): member in TestExperiment.test_torchscript_module_export_various_inputs_with_annotated_input_names.MList
188 def forward(self, x: Dict[str, torch.Tensor]): member in TestExperiment.test_torchscript_module_export_various_inputs_with_annotated_input_names.MDict
202 def forward(self, x): member in TestExperiment.test_joint_basic.Module
[all …]
|
| D | test_serialize.py | 56 def forward(self, x): member in TestSerialize.test_export_with_extension_op_serialization.TestModule
127 def forward(self, x): member in TestSerialize.test_predispatch_export_with_autograd_op.Foo
155 def forward(self, x, y, use_p=False): member in TestSerialize.test_export_example_inputs_preserved.MyModule
189 def forward(self, x): member in TestSerialize.test_metadata_parsing_with_layer_split.MyModule
215 def forward(self, x): member in TestSerialize.test_serialize_constant_outputs.MyModule
237 def forward(self, x, w, b): member in TestSerialize.test_serialize_multiple_returns_from_node.MyModule
273 def forward(self, a, b, c) -> torch.Tensor: member in TestSerialize.test_serialize_sym_int.DynamicShapeSimpleModel
307 def forward(self, x): member in TestSerialize.test_serialize_list_returns.MyModule
349 def forward(self, x): member in TestSerialize.test_multi_return_some_unused.MyModule
371 def forward(self, x): member in TestSerialize.test_rational_ranges.M
[all …]
|
| /external/pytorch/torch/csrc/api/src/nn/modules/ |
| D | pooling.cpp | 25 Tensor AvgPool1dImpl::forward(const Tensor& input) { in forward() function in torch::nn::AvgPool1dImpl
35 Tensor AvgPool2dImpl::forward(const Tensor& input) { in forward() function in torch::nn::AvgPool2dImpl
46 Tensor AvgPool3dImpl::forward(const Tensor& input) { in forward() function in torch::nn::AvgPool3dImpl
79 Tensor MaxPool1dImpl::forward(const Tensor& input) { in forward() function in torch::nn::MaxPool1dImpl
100 Tensor MaxPool2dImpl::forward(const Tensor& input) { in forward() function in torch::nn::MaxPool2dImpl
121 Tensor MaxPool3dImpl::forward(const Tensor& input) { in forward() function in torch::nn::MaxPool3dImpl
148 Tensor AdaptiveMaxPool1dImpl::forward(const Tensor& input) { in forward() function in torch::nn::AdaptiveMaxPool1dImpl
158 Tensor AdaptiveMaxPool2dImpl::forward(const Tensor& input) { in forward() function in torch::nn::AdaptiveMaxPool2dImpl
168 Tensor AdaptiveMaxPool3dImpl::forward(const Tensor& input) { in forward() function in torch::nn::AdaptiveMaxPool3dImpl
190 Tensor AdaptiveAvgPool1dImpl::forward(const Tensor& input) { in forward() function in torch::nn::AdaptiveAvgPool1dImpl
[all …]
|
| D | activation.cpp | 12 Tensor ELUImpl::forward(Tensor input) { in forward() function in torch::nn::ELUImpl
30 Tensor SELUImpl::forward(Tensor input) { in forward() function in torch::nn::SELUImpl
49 Tensor HardshrinkImpl::forward(const Tensor& input) { in forward() function in torch::nn::HardshrinkImpl
68 Tensor HardtanhImpl::forward(Tensor input) { in forward() function in torch::nn::HardtanhImpl
94 Tensor LeakyReLUImpl::forward(Tensor input) { in forward() function in torch::nn::LeakyReLUImpl
112 Tensor LogSigmoidImpl::forward(const Tensor& input) { in forward() function in torch::nn::LogSigmoidImpl
132 Tensor SoftmaxImpl::forward(const Tensor& input) { in forward() function in torch::nn::SoftmaxImpl
146 Tensor SoftminImpl::forward(const Tensor& input) { in forward() function in torch::nn::SoftminImpl
161 Tensor LogSoftmaxImpl::forward(const Tensor& input) { in forward() function in torch::nn::LogSoftmaxImpl
173 Tensor Softmax2dImpl::forward(const Tensor& input) { in forward() function in torch::nn::Softmax2dImpl
[all …]
|
| /external/pytorch/test/quantization/jit/ |
| D | test_quantize_jit.py | 93 def forward(self, x): member in TestQuantizeJitPasses.test_skip_dequant_constant_prop.M
143 def forward(self, x): member in TestQuantizeJitPasses.test_foldbn_trivial.TestModule
188 def forward(self, x): member in TestQuantizeJitPasses.test_foldbn_trivial_nobias.TestModule
229 def forward(self, x): member in TestQuantizeJitPasses.test_foldbn_in_submodule.SubModule
239 def forward(self, x): member in TestQuantizeJitPasses.test_foldbn_in_submodule.TestModule
279 def forward(self, x): member in TestQuantizeJitPasses.test_foldbn_shared_classtype.TestModule
301 def forward(self, x): member in TestQuantizeJitPasses.test_foldbn_no_fusion.CustomConv
305 def forward(self, x): member in TestQuantizeJitPasses.test_foldbn_no_fusion.CustomBn
314 def forward(self, x): member in TestQuantizeJitPasses.test_foldbn_no_fusion.M
347 def forward(self, x): member in TestQuantizeJitPasses.test_foldbn_complex_cases.SubModule
[all …]
|
| /external/pytorch/test/jit/ |
| D | test_module_containers.py | 28 def forward(self, x): member in TestModuleContainers.test_sequential_intermediary_types.A
32 def forward(self, x): member in TestModuleContainers.test_sequential_intermediary_types.B
40 def forward(self, x): member in TestModuleContainers.test_sequential_intermediary_types.C
47 def forward(self, x): member in TestModuleContainers.test_moduledict.Inner
51 def forward(self, x): member in TestModuleContainers.test_moduledict.Inner2
55 def forward(self, x): member in TestModuleContainers.test_moduledict.Inner3
70 def forward(self, x, skip_name): member in TestModuleContainers.test_moduledict.M
93 def forward(self, x, skip_name): member in TestModuleContainers.test_moduledict.M2
131 def forward(self, x): member in TestModuleContainers.test_custom_container_forward.Inner
138 def forward(self, x): member in TestModuleContainers.test_custom_container_forward.CustomSequential
[all …]
|
| D | test_module_interface.py | 32 def forward(self, input: Tensor) -> Tensor: member in OrigModule
40 def forward(self, input: Tensor) -> Tensor: member in NewModule
58 def forward(self, input: Tensor) -> Tensor: member in TestModuleInterface.test_not_submodule_interface_call.TestNotModuleInterfaceCall
75 def forward(self, x: Tensor) -> Tensor: member in TestModuleInterface.test_module_interface.OneTwoModule
93 def forward(self, x: Tensor) -> Tensor: member in TestModuleInterface.test_module_interface.FooMod
103 def forward(self, x: Tensor) -> Tensor: member in TestModuleInterface.test_module_interface.BarMod
159 def forward(self, input): member in TestModuleInterface.test_module_doc_string.TestInterface
173 def forward(self, input): member in TestModuleInterface.test_module_doc_string.TestModule
189 def forward(self, x: Tensor) -> Tensor: member in TestModuleInterface.test_module_interface_subtype.OneTwoModule
206 def forward(self, x: Tensor) -> Tensor: member in TestModuleInterface.test_module_interface_subtype.Foo
[all …]
|
| D | test_freezing.py | 77 def forward(self, x): member in TestFreezing.test_freeze_module.M
135 def forward(self, x): member in TestFreezing.test_freeze_module_with_submodule.SubModule
144 def forward(self, x): member in TestFreezing.test_freeze_module_with_submodule.SubModule2
156 def forward(self, x): member in TestFreezing.test_freeze_module_with_submodule.TestModule
200 def forward(self, x): member in TestFreezing.test_freeze_module_with_fork.SubModule
208 def forward(self, x): member in TestFreezing.test_freeze_module_with_fork.TestModule
240 def forward(self, x): member in TestFreezing.test_freeze_module_with_nested_fork.SubModule
249 def forward(self, x): member in TestFreezing.test_freeze_module_with_nested_fork.SubModule2
261 def forward(self, x): member in TestFreezing.test_freeze_module_with_nested_fork.TestModule
299 def forward(self, x): member in TestFreezing.test_freeze_module_with_fork2.TestModule
[all …]
|
| D | test_scriptmod_ann.py | 37 def forward(self, x: int): member in TestScriptModuleInstanceAttributeTypeAnnotation.test_annotated_falsy_base_type.M
51 def forward(self, x: List[int]): member in TestScriptModuleInstanceAttributeTypeAnnotation.test_annotated_nonempty_container.M
65 def forward(self, x: torch.Tensor): member in TestScriptModuleInstanceAttributeTypeAnnotation.test_annotated_empty_tensor.M
79 def forward(self, x: List[int]): member in TestScriptModuleInstanceAttributeTypeAnnotation.test_annotated_with_jit_attribute.M
95 def forward(self, y: List[int]): member in TestScriptModuleInstanceAttributeTypeAnnotation.test_annotated_class_level_annotation_only.M
111 def forward(self, y: List[int]): member in TestScriptModuleInstanceAttributeTypeAnnotation.test_annotated_class_level_annotation_and_init_annotation.M
127 def forward(self, y: List[int]): member in TestScriptModuleInstanceAttributeTypeAnnotation.test_annotated_class_level_jit_annotation.M
141 def forward(self, x: List[int]): member in TestScriptModuleInstanceAttributeTypeAnnotation.test_annotated_empty_list.M
165 def forward(self, x: list[int]): member in TestScriptModuleInstanceAttributeTypeAnnotation.test_annotated_empty_list_lowercase.M
186 def forward(self, x: Dict[str, int]): member in TestScriptModuleInstanceAttributeTypeAnnotation.test_annotated_empty_dict.M
[all …]
|
| /external/executorch/backends/cadence/aot/tests/ |
| D | test_remove_ops_passes.py | 51 def forward(self, x: torch.Tensor): member in TestRemoveOpsPasses.test_remove_to_ops.M
81 def forward(self, t: torch.Tensor): member in TestRemoveOpsPasses.test_remove_nop_add_op_pass.FullX
85 def forward(self, t: torch.Tensor): member in TestRemoveOpsPasses.test_remove_nop_add_op_pass.FullY
121 def forward(self, t: torch.Tensor): member in TestRemoveOpsPasses.test_remove_nop_mul_op_pass.FullX
125 def forward(self, t: torch.Tensor): member in TestRemoveOpsPasses.test_remove_nop_mul_op_pass.FullY
159 def forward(self, x: torch.Tensor): member in TestRemoveOpsPasses.test_remove_alias_copy.M
184 def forward(self, x: torch.Tensor): member in TestRemoveOpsPasses.test_remove_detach_copy.M
215 def forward(self, x: torch.Tensor): member in TestRemoveOpsPasses.test_remove_zero_sized_constant_pad_nd.Padding
233 def forward(self, x): member in TestRemoveOpsPasses.test_remove_expand.Expand
247 def forward(self, x, y): member in TestRemoveOpsPasses.test_remove_zero_arg_cat.Cat
[all …]
|
| /external/executorch/backends/vulkan/test/ |
| D | test_vulkan_delegate.py | 189 def forward(self, x, y, w): member in TestBackends.test_vulkan_backend_add.AddModule
222 def forward(self, x, y): member in TestBackends.test_vulkan_backend_add_int.AddIntModule
240 def forward(self, x): member in TestBackends.test_vulkan_backend_zero_dim_tensor.ZeroDimModule
253 def forward(self, x, y): member in TestBackends.test_vulkan_backend_internal_data.InternalDataModule
273 def forward(self, x, y): member in TestBackends.test_vulkan_backend_sub.SubModule
292 def forward(self, x, y): member in TestBackends.test_vulkan_backend_mul.MulModule
311 def forward(self, x, y): member in TestBackends.test_vulkan_backend_div.DivModule
331 def forward(self, x, y): member in TestBackends.test_vulkan_backend_arithmetic.ArithmeticModule
351 def forward(self, x, y): member in TestBackends.test_vulkan_backend_floor_div.FloorDivModule
371 def forward(self, x, y): member in TestBackends.test_vulkan_backend_pow.PowModule
[all …]
|
| /external/pytorch/test/nn/ |
| D | test_parametrization.py | 47 def forward(self, X): member in TestNNParametrization.test_register_and_remove_parametrization.Skew
52 def forward(self, X): member in TestNNParametrization.test_register_and_remove_parametrization.Orthogonal
63 def forward(self, X): member in TestNNParametrization.test_register_and_remove_parametrization.Resize
67 def forward(self, X): member in TestNNParametrization.test_register_and_remove_parametrization.NoResize
72 def forward(self, x): member in TestNNParametrization.test_register_and_remove_parametrization.FirstZero
76 def forward(self, x): member in TestNNParametrization.test_register_and_remove_parametrization.LastZero
299 def forward(self, X): member in TestNNParametrization.test_register_and_remove_nested_parametrization.Skew
352 def forward(self, x): member in TestNNParametrization.test_register_and_remove_buffer_parametrization.FirstZero
356 def forward(self, x): member in TestNNParametrization.test_register_and_remove_buffer_parametrization.LastZero
397 def forward(self, X): member in TestNNParametrization.test_serialization_parametrization.Orthogonal
[all …]
|
| /external/executorch/backends/apple/mps/test/ |
| D | test_mps.py | 179 def forward(self, x): member in TestMPSUnitOpTesting.test_mps_backend_split_copy.SplitCopy
192 def forward(self, x): member in TestMPSUnitOpTesting.test_mps_backend_unbind_copy.UnbindCopy
205 def forward(self, x): member in TestMPSUnitOpTesting.test_mps_backend_pixel_shuffle.PixelShuffle
218 def forward(self, *x): member in TestMPSUnitOpTesting.test_mps_backend_cumsum.CumulativeSum
231 def forward(self, *x): member in TestMPSUnitOpTesting.test_mps_backend_stack.Stack
247 def forward(self, *x): member in TestMPSUnitOpTesting.test_mps_backend_cat.Cat
264 def forward(self, x): member in TestMPSUnitOpTesting.test_mps_backend_expand_copy.ExpandCopy
277 def forward(self, x): member in TestMPSUnitOpTesting.test_mps_backend_select.Select
291 def forward(self, x): member in TestMPSUnitOpTesting.test_mps_backend_view_copy.ViewCopy
304 def forward(self, x): member in TestMPSUnitOpTesting.test_mps_backend_mean_dim_2.Mean
[all …]
|
| /external/pytorch/test/dynamo/ |
| D | test_modules.py | 52 def forward(self, x): member in BasicModule
62 def forward(self, x): member in FnMember
75 def forward(self, x): member in FnMemberCmp
91 def forward(self, x): member in SubmoduleExample
104 def forward(self, x): member in IsTrainingCheck
129 def forward(self, x): member in ModuleMethodCall
146 def forward(self, x): member in UnsupportedMethodCall
157 def forward(self, x): member in UnsupportedModule
167 def forward(self, x): member in UnsupportedModuleCall
173 def forward(x): member in ModuleWithStaticForward
[all …]
|
| /external/pytorch/test/jit/fixtures_srcs/ |
| D | fixtures_src.py | 7 def forward(self, a, b): member in TestVersionedDivTensorExampleV7
15 def forward(self, a: Union[int, float, complex], b: Union[int, float, complex]): member in TestVersionedLinspaceV7
22 def forward( member in TestVersionedLinspaceOutV7
32 def forward(self, a: Union[int, float, complex], b: Union[int, float, complex]): member in TestVersionedLogspaceV8
39 def forward( member in TestVersionedLogspaceOutV8
49 def forward(self, x): member in TestVersionedGeluV9
54 def forward(self, x): member in TestVersionedGeluOutV9
60 def forward(self, x): member in TestVersionedRandomV10
66 def forward(self, x): member in TestVersionedRandomFuncV10
72 def forward(self, x): member in TestVersionedRandomOutV10
|
| /external/pytorch/test/fx/ |
| D | test_z3_gradual_types.py | 54 def forward(self, x: TensorType([1, 2])): member in TorchDynamoUseCases.test_dim.BasicBlock
73 def forward(self, x: Dyn): member in TorchDynamoUseCases.test_reshape.BasicBlock
96 def forward(self, x: TensorType([32, 4, 4])): member in HFOperations.test_eq_dim.BasicBlock
125 def forward(self, x: TensorType([32, 4, 4]), y: TensorType([32, 4, 4])): member in HFOperations.test_conditional_ne_1.BasicBlock
151 def forward(self, x: TensorType([Dyn, 2, 3]), y: TensorType([1, 3, 2])): member in HFOperations.test_bmm.BasicBlock
170 def forward(self, x: Dyn, y: TensorType([1, 3, 2])): member in HFOperations.test_bmm2.BasicBlock
189 def forward(self, x: TensorType([2, 3, 3]), y: TensorType([1, 3, 2])): member in HFOperations.test_bmm3.BasicBlock
202 def forward(self, x: TensorType([1, 2, 3, 4])): member in HFOperations.test_transpose.BasicBlock
233 def forward(self, x: TensorType([2050, 1024]), y: Dyn): member in HFOperations.test_index_select.BasicBlock
265 def forward(self, x: TensorType([1, 2, 3])): member in HFOperations.test_get_attr.BasicBlock
[all …]
|