| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_quantization.py | 1361 def get_example_inputs(self) -> Tuple[Any, ...]: member in SingleLayerLinearModel
1374 def get_example_inputs(self) -> Tuple[Any, ...]: member in AnnotatedSingleLayerLinearModel
1387 def get_example_inputs(self) -> Tuple[Any, ...]: member in SingleLayerLinearDynamicModel
1402 def get_example_inputs(self) -> Tuple[Any, ...]: member in LinearAddModel
1454 def get_example_inputs(self) -> Tuple[Any, ...]: member in ConvModel
1466 def get_example_inputs(self) -> Tuple[Any, ...]: member in ConvTransposeModel
1483 def get_example_inputs(self) -> Tuple[Any, ...]: member in AnnotatedConvModel
1500 def get_example_inputs(self) -> Tuple[Any, ...]: member in AnnotatedConvTransposeModel
1514 def get_example_inputs(self) -> Tuple[Any, ...]: member in ConvBnModel
1533 def get_example_inputs(self) -> Tuple[Any, ...]: member in AnnotatedConvBnModel
[all …]
|
| /aosp_15_r20/external/executorch/examples/models/toy_model/ |
| H A D | model.py | 23 def get_example_inputs(self): member in MulModule
38 def get_example_inputs(self): member in LinearModule
53 def get_example_inputs(self): member in AddModule
69 def get_example_inputs(self): member in AddMulModule
89 def get_example_inputs(self): member in SoftmaxModule
|
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_layout_optim.py | 32 def get_example_inputs(self): member in Model2Conv
149 def get_example_inputs(self): member in TestLayoutOptim.test_3conv_with_graph_break.Model
167 def get_example_inputs(self): member in TestLayoutOptim.test_keep_output_layout_infer.Model
245 def get_example_inputs(self): member in TestLayoutOptim.test_mutate_base_for_conv_output.Model
263 def get_example_inputs(self): member in TestLayoutOptim.test_mutate_view_for_conv_output.Model
|
| /aosp_15_r20/external/executorch/backends/xnnpack/test/models/ |
| H A D | emformer_rnnt.py | 27 def get_example_inputs(self): member in TestEmformerModel.Joiner
52 def get_example_inputs(self): member in TestEmformerModel.Predictor
78 def get_example_inputs(self): member in TestEmformerModel.Transcriber
|
| /aosp_15_r20/external/executorch/examples/models/emformer_rnnt/ |
| H A D | model.py | 54 def get_example_inputs(self): member in EmformerRnntTranscriberModel
89 def get_example_inputs(self): member in EmformerRnntPredictorModel
125 def get_example_inputs(self): member in EmformerRnntJoinerModel
|
| /aosp_15_r20/external/pytorch/test/quantization/fx/ |
| H A D | test_model_report_fx.py | 99 def get_example_inputs(self): member in ThreeOps
115 def get_example_inputs(self): member in TwoThreeOps
1291 def get_example_inputs(self): member in TestFxDetectInputWeightEqualization.TwoBlockComplexNet
1303 def get_example_inputs(self): member in TestFxDetectInputWeightEqualization.ReluOnly
1519 def get_example_inputs(self): member in TestFxDetectOutliers.LargeBatchModel
|
| /aosp_15_r20/external/executorch/examples/models/deeplab_v3/ |
| H A D | model.py | 31 def get_example_inputs(self): member in DeepLabV3ResNet50Model
48 def get_example_inputs(self): member in DeepLabV3ResNet101Model
|
| /aosp_15_r20/external/executorch/examples/models/resnet/ |
| H A D | model.py | 31 def get_example_inputs(self): member in ResNet18Model
46 def get_example_inputs(self): member in ResNet50Model
|
| /aosp_15_r20/external/executorch/examples/models/mobilenet_v2/ |
| H A D | model.py | 27 def get_example_inputs(self): member in MV2Model
40 def get_example_inputs(self): member in MV2UntrainedModel
|
| /aosp_15_r20/external/executorch/examples/models/edsr/ |
| H A D | model.py | 26 def get_example_inputs(self): member in EdsrModel
|
| /aosp_15_r20/external/executorch/examples/models/ |
| H A D | model_base.py | 40 def get_example_inputs(self): member in EagerModelBase
|
| /aosp_15_r20/external/executorch/examples/models/wav2letter/ |
| H A D | model.py | 27 def get_example_inputs(self): member in Wav2LetterModel
|
| /aosp_15_r20/external/executorch/examples/models/inception_v3/ |
| H A D | model.py | 26 def get_example_inputs(self): member in InceptionV3Model
|
| /aosp_15_r20/external/executorch/examples/models/lstm/ |
| H A D | model.py | 27 def get_example_inputs(self): member in LSTMModel
|
| /aosp_15_r20/external/executorch/examples/models/torchvision_vit/ |
| H A D | model.py | 26 def get_example_inputs(self): member in TorchVisionViTModel
|
| /aosp_15_r20/external/executorch/examples/models/mobilenet_v3/ |
| H A D | model.py | 28 def get_example_inputs(self): member in MV3Model
|
| /aosp_15_r20/external/executorch/examples/models/mobilebert/ |
| H A D | model.py | 29 def get_example_inputs(self): member in MobileBertModelExample
|
| /aosp_15_r20/external/executorch/examples/models/inception_v4/ |
| H A D | model.py | 25 def get_example_inputs(self): member in InceptionV4Model
|
| /aosp_15_r20/external/executorch/backends/example/ |
| H A D | test_example_delegate.py | 38 def get_example_inputs(): member in TestExampleDelegate.test_delegate_linear.Conv2dModule
|
| /aosp_15_r20/external/executorch/backends/apple/mps/test/ |
| H A D | test_mps_models.py | 23 def get_example_inputs(): member in LayerNormModule
36 def get_example_inputs(): member in Conv2DModule
|
| /aosp_15_r20/external/executorch/examples/models/llama3_2_vision/text_decoder/ |
| H A D | model.py | 164 def get_example_inputs(self): member in Llama3_2Decoder
|
| /aosp_15_r20/external/executorch/examples/models/llama/ |
| H A D | model.py | 242 def get_example_inputs(self): member in Llama2Model
|
| /aosp_15_r20/external/executorch/exir/tests/ |
| H A D | test_remove_view_copy.py | 38 def get_example_inputs(self): member in TestModel1
|
| /aosp_15_r20/external/executorch/examples/models/llama3_2_vision/preprocess/ |
| H A D | model.py | 57 def get_example_inputs(self) -> Tuple[torch.Tensor, torch.Tensor, torch.Tensor]: member in CLIPImageTransformModel
|
| /aosp_15_r20/external/executorch/examples/mediatek/models/llm_models/ |
| H A D | modeling_base.py | 37 def get_example_inputs(self, num_token, cache_size, get_dym_shape): member in BaseModelChunk
|