| /aosp_15_r20/external/pytorch/test/onnx/ |
| H A 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 …]
|
| H A 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 …]
|
| H A 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 …]
|
| /aosp_15_r20/external/pytorch/test/export/ |
| H A 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 …]
|
| H A 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 …]
|
| H A 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 …]
|
| H A 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 …]
|
| /aosp_15_r20/external/executorch/backends/qualcomm/tests/ |
| H A 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 …]
|
| /aosp_15_r20/external/pytorch/test/quantization/jit/ |
| H A 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 …]
|
| /aosp_15_r20/external/pytorch/test/jit/ |
| H A 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 …]
|
| H A 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 …]
|
| H A 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 …]
|
| /aosp_15_r20/external/executorch/backends/apple/mps/test/ |
| H A 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 …]
|
| /aosp_15_r20/external/executorch/backends/vulkan/test/ |
| H A 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 …]
|
| /aosp_15_r20/external/pytorch/test/dynamo/ |
| H A 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 …]
|
| H A D | test_autograd_function.py | 22 def forward(ctx, foo): member in CustomFunc1
33 def forward(ctx, foo): member in CustomFunc3
47 def forward(self, foo): member in Module1
56 def forward(self, foo): member in Module2
61 def forward(self, foo): member in Module3
70 def forward(self, foo): member in Module4
75 def forward(self, foo): member in Module5
84 def forward(self, foo): member in Module6
91 def forward(input, weight, bias): member in LinearFunction
120 def forward(self, input, weight, bias=None): member in ModuleLinear
[all …]
|
| /aosp_15_r20/external/pytorch/test/quantization/fx/ |
| H A D | test_quantize_fx.py | 218 def forward(self, x, y): member in BinaryOp
235 def forward(self, x, y): member in BinaryOpNonQuantizedInput
256 def forward(self, x, y): member in BinaryOpRelu
288 def forward(self, x): member in TestFuseFx.test_fuse_conv_bn_relu.M
357 def forward(self, x): member in TestFuseFx.test_fuse_linear_bn_eval.M
600 def forward(self, x): member in TestFuseFx.test_fuse_module_relu.M
677 def forward(self, x): member in TestFuseFx.test_problematic_fuse_example.M
710 def forward(self, x): member in TestFuseFx.test_fuse_addtional_fuser_method.M
734 def forward(self, x): member in TestFuseFx.test_fuse_custom_pattern.M
782 def forward(self, x): member in TestFuseFx.test_fusion_pattern_with_multiple_inputs.M
[all …]
|
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_aot_inductor.py | 181 def forward(self, x, y): member in AOTInductorTestsTemplate.test_simple.Model
196 def forward(self, x): member in AOTInductorTestsTemplate.test_small_constant.Model
209 def forward(self, x, y): member in AOTInductorTestsTemplate.test_output_path_1.Model
225 def forward(self, x, y): member in AOTInductorTestsTemplate.test_output_path_2.Model
246 def forward(self, x): member in AOTInductorTestsTemplate.test_constant_folding.Model
266 def forward(self, x): member in AOTInductorTestsTemplate.test_duplicate_constant_folding.Model
277 def forward(self, x): member in AOTInductorTestsTemplate.test_multi_device.Model
293 def forward(self, x, y): member in AOTInductorTestsTemplate.test_large_weight.Model
312 def forward(self, x, y): member in AOTInductorTestsTemplate.test_large_mmaped_weights.Model
329 def forward(self, x, y): member in AOTInductorTestsTemplate.test_with_offset.Model
[all …]
|
| /aosp_15_r20/external/pytorch/test/fx/ |
| H A 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 …]
|
| /aosp_15_r20/external/pytorch/test/nn/ |
| H A 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 …]
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_fx.py | 75 def forward(self, x): member in SimpleTest
159 def forward(self, x): member in Add
200 def forward(self, x): member in TestFX.test_graph_module.MySub
210 def forward(self, A, B, c): member in TestFX.test_graph_module.MyModule
220 def forward(self, A): member in TestFX.test_graph_module.M2
229 def forward(self, A, b=4, *args, c=5, **kwargs): member in TestFX.test_graph_module.T
238 def forward(self, x): member in TestFX.test_graph_module.M3
251 def forward(self, a): member in TestFX.test_informative_co_filename.MyModule
270 def forward(self, *args, **kwargs): member in TestFX.test_args_kwargs.T
279 def forward(self, *args, **kwargs): member in TestFX.test_varargs_concrete.T
[all …]
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/modules/ |
| H A 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 …]
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_quantization.py | 1340 def forward(self, x): member in PT2EQuantizationTestCase._get_pt2e_quantized_linear.M
1357 def forward(self, x): member in SingleLayerLinearModel
1370 def forward(self, x): member in AnnotatedSingleLayerLinearModel
1383 def forward(self, x): member in SingleLayerLinearDynamicModel
1396 def forward(self, x): member in LinearAddModel
1414 def forward(self, x): member in RNNDynamicModel
1431 def forward(self, x): member in RNNCellDynamicModel
1441 def forward(self, x, hid): member in LSTMwithHiddenDynamicModel
1450 def forward(self, x): member in ConvModel
1462 def forward(self, x): member in ConvTransposeModel
[all …]
|
| /aosp_15_r20/external/executorch/backends/cadence/aot/tests/ |
| H A 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 …]
|
| /aosp_15_r20/external/executorch/exir/tests/ |
| H A D | test_passes.py | 126 def forward(self, x: torch.Tensor, y: torch.Tensor) -> torch.Tensor: member in TestPasses.test_remove_mixed_type_operators.Add
182 def forward(self, x: torch.Tensor, y: torch.Tensor) -> torch.Tensor: member in TestPasses.test_remove_mixed_type_operators.Mult
217 def forward(self, x: torch.Tensor) -> torch.Tensor: member in TestPasses.test_remove_noop_pass.Foo
238 def forward(self, x: torch.Tensor) -> torch.Tensor: member in TestPasses.test_redundant_slice_copy_removal.FooWithNoSlice
244 def forward(self, x: torch.Tensor) -> torch.Tensor: member in TestPasses.test_redundant_slice_copy_removal.FooWithOneSlice
250 def forward(self, x: torch.Tensor) -> torch.Tensor: member in TestPasses.test_redundant_slice_copy_removal.FooWithAllSlices
295 def forward(self, x: torch.Tensor) -> torch.Tensor: member in TestPasses.test_compile_to_edge.Foo
322 def forward(self, x_raw, h, c): member in TestPasses.test_to_out_variant_none_output.CompositeModel
371 def forward(self, x): member in TestPasses.test_to_out_variant_singleon_tensor_list.MyModel
406 def forward(self, x): member in TestPasses.test_to_out_variant_multiple_out.MyModel
[all …]
|