Home
last modified time | relevance | path

Searched refs:create_model (Results 1 – 25 of 41) sorted by relevance

12

/aosp_15_r20/external/pytorch/test/onnx/
H A Dtest_fx_to_onnx_with_onnxruntime.py722 create_model: Callable,
748 model = create_model()
770 fake_model = create_model()
861 def create_model() -> nn.Module: function
872 create_model,
884 def create_model() -> nn.Module: function
899 create_model,
948 create_model: Callable,
981 real_model = create_model()
994 fake_model = create_model()
[all …]
/aosp_15_r20/external/executorch/examples/models/test/
H A Dtest_export.py72 eager_model, example_inputs, _, _ = EagerModelFactory.create_model(
84 eager_model, example_inputs, _, _ = EagerModelFactory.create_model(
93 eager_model, example_inputs, _, _ = EagerModelFactory.create_model(
105 eager_model, example_inputs, _, _ = EagerModelFactory.create_model(
114 eager_model, example_inputs, _, _ = EagerModelFactory.create_model(
126 eager_model, example_inputs, _, _ = EagerModelFactory.create_model(
135 eager_model, example_inputs, _, _ = EagerModelFactory.create_model(
144 eager_model, example_inputs, _, _ = EagerModelFactory.create_model(
/aosp_15_r20/external/tensorflow/tensorflow/examples/speech_commands/
H A Dmodels_test.py52 logits, dropout_rate = models.create_model(
64 logits = models.create_model(fingerprint_input, model_settings, "conv",
74 logits, dropout_rate = models.create_model(
86 logits, dropout_rate = models.create_model(
98 models.create_model(fingerprint_input, model_settings,
107 logits, dropout_rate = models.create_model(
/aosp_15_r20/external/libaom/test/
H A Dav1_external_partition_test.cc297 ext_part_funcs.create_model = ext_part_create_model; in PreEncodeFrameHook()
329 ext_part_funcs.create_model = ext_part_create_model; in PreEncodeFrameHook()
606 ext_part_funcs.create_model = ext_part_create_model; in PreEncodeFrameHook()
610 ext_part_funcs.create_model = ext_part_create_model; in PreEncodeFrameHook()
613 ext_part_funcs.create_model = ext_part_create_model_test; in PreEncodeFrameHook()
/aosp_15_r20/external/pytorch/test/distributed/fsdp/
H A Dtest_fsdp_memory.py93 def create_model(with_fsdp, with_checkpoint, model_hidden_dim): function
115 model = create_model(
167 model = create_model(
/aosp_15_r20/external/pytorch/benchmarks/distributed/ddp/
H A Dbenchmark.py54 model = benchmark.create_model()
162 def create_model(self): member in Benchmark
184 def create_model(self): member in TorchvisionBenchmark
/aosp_15_r20/external/executorch/examples/portable/scripts/
H A Dexport_and_delegate.py60 m, m_inputs, _, _ = EagerModelFactory.create_model(*MODEL_NAME_TO_MODEL["add_mul"])
169 m, m_inputs, _, _ = EagerModelFactory.create_model(*MODEL_NAME_TO_MODEL["add_mul"])
H A Dexport.py61 model, example_inputs, _, dynamic_shapes = EagerModelFactory.create_model(
/aosp_15_r20/external/executorch/examples/qualcomm/oss_scripts/
H A Dfastvit.py48 from timm.models import create_model
51 model = create_model("fastvit_s12")
H A DgMLP_image_classification.py48 model = timm.create_model("gmlp_s16_224", pretrained=True).eval()
H A Dfbnet.py35 instance = timm.create_model("fbnetc_100", pretrained=True).eval()
/aosp_15_r20/external/mesa3d/src/gallium/targets/teflon/
H A Dtest_model_generation.py138 def create_model(batch_size, in_w, in_h, k_w, k_h, in_ch, out_ch, stride, padding, signed, seed, de… function
173 …model_file_2 = create_model(batch_size, in_w, in_h, k_w, k_h, in_ch, out_ch, stride, padding, sign…
/aosp_15_r20/external/pytorch/benchmarks/dynamo/
H A Dtimm_models.py39 from timm.models import create_model
248 model = create_model(
/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_padding.py199 def create_model(vocab_size): function
218 f_good_shape, inputs_good_shape = create_model(30528)
219 f_bad_shape, inputs_bad_shape = create_model(30522)
/aosp_15_r20/external/executorch/examples/models/
H A Dmodel_factory.py20 def create_model( member in EagerModelFactory
/aosp_15_r20/external/executorch/examples/devtools/scripts/
H A Dgen_sample_etrecord.py77 model, example_inputs, _, _ = EagerModelFactory.create_model(
H A Dexport_bundled_program.py142 model, example_inputs, _, _ = EagerModelFactory.create_model(
/aosp_15_r20/external/executorch/examples/xnnpack/
H A Daot_compiler.py82 model, example_inputs, _, _ = EagerModelFactory.create_model(
/aosp_15_r20/external/libaom/av1/encoder/
H A Dexternal_partition.c24 const aom_ext_part_status_t status = ext_part_controller->funcs.create_model( in av1_ext_part_create()
/aosp_15_r20/external/executorch/examples/qualcomm/scripts/
H A Dexport_example.py56 model, example_inputs, _, _ = EagerModelFactory.create_model(
/aosp_15_r20/external/libaom/aom/
H A Daom_external_partition.h414 aom_ext_part_create_model_fn_t create_model; member
/aosp_15_r20/external/libvpx/vpx/
H A Dvpx_ext_ratectrl.h542 vpx_rc_create_model_cb_fn_t create_model; member
/aosp_15_r20/external/executorch/examples/apple/coreml/scripts/
H A Ddebugger_cli.py152 model, example_inputs, _, _ = EagerModelFactory.create_model(
/aosp_15_r20/external/executorch/examples/xnnpack/quantization/
H A Dexample.py165 model, example_inputs, _, _ = EagerModelFactory.create_model(
/aosp_15_r20/external/executorch/examples/apple/mps/scripts/
H A Dmps_example.py155 model, example_inputs, _, _ = EagerModelFactory.create_model(**model_config)

12