Home
last modified time | relevance | path

Searched defs:get_example_inputs (Results 1 – 25 of 38) sorted by relevance

12

/external/executorch/examples/models/toy_model/
Dmodel.py23 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
/external/executorch/backends/xnnpack/test/models/
Demformer_rnnt.py27 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
/external/executorch/examples/models/resnet/
Dmodel.py31 def get_example_inputs(self): member in ResNet18Model
46 def get_example_inputs(self): member in ResNet50Model
/external/executorch/examples/models/deeplab_v3/
Dmodel.py31 def get_example_inputs(self): member in DeepLabV3ResNet50Model
48 def get_example_inputs(self): member in DeepLabV3ResNet101Model
/external/executorch/examples/models/mobilenet_v2/
Dmodel.py27 def get_example_inputs(self): member in MV2Model
40 def get_example_inputs(self): member in MV2UntrainedModel
/external/executorch/examples/models/emformer_rnnt/
Dmodel.py54 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
/external/pytorch/test/inductor/
Dtest_layout_optim.py32 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
/external/pytorch/torch/testing/_internal/
Dcommon_quantization.py1361 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 …]
/external/executorch/examples/models/mobilebert/
Dmodel.py29 def get_example_inputs(self): member in MobileBertModelExample
/external/executorch/examples/models/edsr/
Dmodel.py26 def get_example_inputs(self): member in EdsrModel
/external/executorch/examples/models/inception_v3/
Dmodel.py26 def get_example_inputs(self): member in InceptionV3Model
/external/executorch/examples/models/
Dmodel_base.py40 def get_example_inputs(self): member in EagerModelBase
/external/executorch/examples/models/inception_v4/
Dmodel.py25 def get_example_inputs(self): member in InceptionV4Model
/external/executorch/examples/models/torchvision_vit/
Dmodel.py26 def get_example_inputs(self): member in TorchVisionViTModel
/external/executorch/examples/models/wav2letter/
Dmodel.py27 def get_example_inputs(self): member in Wav2LetterModel
/external/executorch/examples/models/mobilenet_v3/
Dmodel.py28 def get_example_inputs(self): member in MV3Model
/external/executorch/examples/models/lstm/
Dmodel.py27 def get_example_inputs(self): member in LSTMModel
/external/executorch/backends/example/
Dtest_example_delegate.py38 def get_example_inputs(): member in TestExampleDelegate.test_delegate_linear.Conv2dModule
/external/executorch/examples/models/llama3_2_vision/text_decoder/
Dmodel.py164 def get_example_inputs(self): member in Llama3_2Decoder
/external/executorch/examples/models/llama/
Dmodel.py242 def get_example_inputs(self): member in Llama2Model
/external/executorch/exir/tests/
Dtest_remove_view_copy.py38 def get_example_inputs(self): member in TestModel1
/external/pytorch/test/quantization/fx/
Dtest_model_report_fx.py99 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
/external/executorch/examples/models/llama3_2_vision/preprocess/
Dmodel.py57 def get_example_inputs(self) -> Tuple[torch.Tensor, torch.Tensor, torch.Tensor]: member in CLIPImageTransformModel
/external/executorch/examples/mediatek/models/llm_models/
Dmodeling_base.py37 def get_example_inputs(self, num_token, cache_size, get_dym_shape): member in BaseModelChunk
/external/executorch/examples/models/llama3_2_vision/vision_encoder/
Dmodel.py81 def get_example_inputs(self): member in FlamingoVisionEncoderModel

12