/aosp_15_r20/external/pytorch/test/onnx/ |
H A D | test_fx_to_onnx_with_onnxruntime.py | 722 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 D | test_export.py | 72 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 D | models_test.py | 52 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 D | av1_external_partition_test.cc | 297 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 D | test_fsdp_memory.py | 93 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 D | benchmark.py | 54 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 D | export_and_delegate.py | 60 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 D | export.py | 61 model, example_inputs, _, dynamic_shapes = EagerModelFactory.create_model(
|
/aosp_15_r20/external/executorch/examples/qualcomm/oss_scripts/ |
H A D | fastvit.py | 48 from timm.models import create_model 51 model = create_model("fastvit_s12")
|
H A D | gMLP_image_classification.py | 48 model = timm.create_model("gmlp_s16_224", pretrained=True).eval()
|
H A D | fbnet.py | 35 instance = timm.create_model("fbnetc_100", pretrained=True).eval()
|
/aosp_15_r20/external/mesa3d/src/gallium/targets/teflon/ |
H A D | test_model_generation.py | 138 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 D | timm_models.py | 39 from timm.models import create_model 248 model = create_model(
|
/aosp_15_r20/external/pytorch/test/inductor/ |
H A D | test_padding.py | 199 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 D | model_factory.py | 20 def create_model( member in EagerModelFactory
|
/aosp_15_r20/external/executorch/examples/devtools/scripts/ |
H A D | gen_sample_etrecord.py | 77 model, example_inputs, _, _ = EagerModelFactory.create_model(
|
H A D | export_bundled_program.py | 142 model, example_inputs, _, _ = EagerModelFactory.create_model(
|
/aosp_15_r20/external/executorch/examples/xnnpack/ |
H A D | aot_compiler.py | 82 model, example_inputs, _, _ = EagerModelFactory.create_model(
|
/aosp_15_r20/external/libaom/av1/encoder/ |
H A D | external_partition.c | 24 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 D | export_example.py | 56 model, example_inputs, _, _ = EagerModelFactory.create_model(
|
/aosp_15_r20/external/libaom/aom/ |
H A D | aom_external_partition.h | 414 aom_ext_part_create_model_fn_t create_model; member
|
/aosp_15_r20/external/libvpx/vpx/ |
H A D | vpx_ext_ratectrl.h | 542 vpx_rc_create_model_cb_fn_t create_model; member
|
/aosp_15_r20/external/executorch/examples/apple/coreml/scripts/ |
H A D | debugger_cli.py | 152 model, example_inputs, _, _ = EagerModelFactory.create_model(
|
/aosp_15_r20/external/executorch/examples/xnnpack/quantization/ |
H A D | example.py | 165 model, example_inputs, _, _ = EagerModelFactory.create_model(
|
/aosp_15_r20/external/executorch/examples/apple/mps/scripts/ |
H A D | mps_example.py | 155 model, example_inputs, _, _ = EagerModelFactory.create_model(**model_config)
|