Home
last modified time | relevance | path

Searched defs:get_random_inputs (Results 1 – 15 of 15) sorted by relevance

/aosp_15_r20/external/executorch/backends/apple/mps/test/
H A Dtest_mps_models.py47 def get_random_inputs(self): member in ModuleBasic
59 def get_random_inputs(self): member in ModuleOpsReturnMulti
70 def get_random_inputs(self): member in ModuleAdd
81 def get_random_inputs(self): member in ModuleFloatAddWithAlpha
92 def get_random_inputs(self): member in ModuleIntAddWithAlpha
109 def get_random_inputs(self): member in ModuleContainers
124 def get_random_inputs(self): member in ToyModelForMemPlanning
142 def get_random_inputs(self): member in MemPlanningWithScratchTensor
157 def get_random_inputs(self): member in ModuleOpsReturnTensorList
168 def get_random_inputs(self): member in ModuleReturnInput
[all …]
/aosp_15_r20/external/executorch/test/end2end/
H A Dtest_end2end.py99 def get_random_inputs(self): member in ModuleBasic
111 def get_random_inputs(self): member in ModuleOpsReturnMulti
122 def get_random_inputs(self): member in ModuleAdd
133 def get_random_inputs(self): member in ModuleFloatAddWithAlpha
144 def get_random_inputs(self): member in ModuleIntAddWithAlpha
161 def get_random_inputs(self): member in ModuleContainers
176 def get_random_inputs(self): member in ToyModelForMemPlanning
194 def get_random_inputs(self): member in MemPlanningWithScratchTensor
209 def get_random_inputs(self): member in ModuleOpsReturnTensorList
220 def get_random_inputs(self): member in ModuleReturnInput
[all …]
H A Dexported_module.py59 def get_random_inputs(self) -> Sequence: member in ExportedModule
/aosp_15_r20/external/executorch/test/models/
H A Dexport_program.py43 def get_random_inputs(self): member in ModuleBasic
65 def get_random_inputs(self): member in ModuleIndex
76 def get_random_inputs(self): member in ModuleNoOp
87 def get_random_inputs(self): member in ModuleAdd
98 def get_random_inputs(self): member in ModuleAddHalf
116 def get_random_inputs(self): member in ModuleDynamicCatUnallocatedIO
144 def get_random_inputs(self): member in ModuleLinear
160 def get_random_inputs(self): member in ModuleMultipleEntry
177 def get_random_inputs(self): member in ModuleSimpleTrain
H A Dexport_delegated_program.py51 def get_random_inputs(self) -> Sequence[torch.Tensor]: member in ModuleAddMul
/aosp_15_r20/external/executorch/exir/tests/
H A Dmodels.py37 def get_random_inputs(self) -> Tuple[torch.Tensor, torch.Tensor]: member in TensorItem
51 def get_random_inputs(self) -> Tuple[torch.Tensor, torch.Tensor]: member in Repeat
67 def get_random_inputs(self) -> Tuple[torch.Tensor, torch.Tensor]: member in ModelWithUnusedArg
91 def get_random_inputs(self) -> Tuple[torch.Tensor, ...]: member in MLP
134 def get_random_inputs(self) -> Tuple[Tensor, Tensor]: member in Mul
145 def get_random_inputs(self) -> Tuple[Tensor, Tensor]: member in ElementwiseAdd
156 def get_random_inputs(self) -> Tuple[Tensor]: member in BasicSinMax
171 def get_random_inputs(self) -> Tuple[Tensor, Tensor]: member in CompositeDelegateModule.__init__.DelegateAdd
194 def get_random_inputs(self) -> Tuple[Tensor, Tensor, Tensor]: member in CompositeDelegateModule
504 def get_random_inputs(self) -> Tuple[Tensor, Tensor, Tensor, Tensor, Tensor]: member in Emformer
H A Dcontrol_flow_models.py26 def get_random_inputs(self): member in FTCondBasic
48 def get_random_inputs(self): member in FTCondDynShape
78 def get_random_inputs(self): member in FTCondDeadCode
92 def get_random_inputs(self): member in FTMapBasic
109 def get_random_inputs(self): member in FTMapDynShape
H A Dtest_memory_planning.py84 def get_random_inputs(self) -> Tuple[torch.Tensor, ...]: member in ToyModelForMemPlanning
104 def get_random_inputs(self) -> Tuple[torch.Tensor, ...]: member in ModelWithDifferentTensorSizes
119 def get_random_inputs(self) -> Tuple[torch.Tensor, ...]: member in ModuleReturnTwo
138 def get_random_inputs(self) -> Tuple[torch.Tensor, ...]: member in ModuleListArg
H A Dtest_serde.py100 def get_random_inputs(self): member in TestSerde.test_to_out_variant_singleon_tensor_list.MyModel
121 def get_random_inputs(self): member in TestSerde.test_to_out_variant_multiple_out.MyModel
H A Ddynamic_shape_models.py31 def get_random_inputs(self): member in BatchNormModel
H A Dasr_joiner.py76 def get_random_inputs(self): member in ASRJoiner
H A Dtransformer.py110 def get_random_inputs(self, method): member in Transformer
H A Dtest_passes.py374 def get_random_inputs(self): member in TestPasses.test_to_out_variant_singleon_tensor_list.MyModel
409 def get_random_inputs(self): member in TestPasses.test_to_out_variant_multiple_out.MyModel
H A Dtest_tracer.py133 def get_random_inputs(self) -> Tuple[torch.Tensor, ...]: member in TestTorchDispatchFXTracer.test_tensor_spec_for_const_tensors.Module
/aosp_15_r20/external/executorch/extension/training/pybindings/test/
H A Dtest.py31 def get_random_inputs(self): member in TestTraining.ModuleSimpleTrain