/aosp_15_r20/external/executorch/backends/apple/mps/test/ |
H A D | test_mps_models.py | 47 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 D | test_end2end.py | 99 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 D | exported_module.py | 59 def get_random_inputs(self) -> Sequence: member in ExportedModule
|
/aosp_15_r20/external/executorch/test/models/ |
H A D | export_program.py | 43 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 D | export_delegated_program.py | 51 def get_random_inputs(self) -> Sequence[torch.Tensor]: member in ModuleAddMul
|
/aosp_15_r20/external/executorch/exir/tests/ |
H A D | models.py | 37 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 D | control_flow_models.py | 26 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 D | test_memory_planning.py | 84 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 D | test_serde.py | 100 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 D | dynamic_shape_models.py | 31 def get_random_inputs(self): member in BatchNormModel
|
H A D | asr_joiner.py | 76 def get_random_inputs(self): member in ASRJoiner
|
H A D | transformer.py | 110 def get_random_inputs(self, method): member in Transformer
|
H A D | test_passes.py | 374 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 D | test_tracer.py | 133 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 D | test.py | 31 def get_random_inputs(self): member in TestTraining.ModuleSimpleTrain
|