Home
last modified time | relevance | path

Searched defs:create_model (Results 1 – 12 of 12) sorted by relevance

/aosp_15_r20/external/pytorch/test/onnx/
H A Dtest_fx_to_onnx_with_onnxruntime.py861 def create_model() -> nn.Module: function
884 def create_model() -> nn.Module: function
1070 def create_model() -> nn.Module: function
1111 def create_model() -> nn.Module: function
1153 def create_model() -> nn.Module: function
1191 def create_model(): function
1234 def create_model() -> nn.Module: function
1280 def create_model(): function
1309 def create_model(): function
1331 def create_model(): function
[all …]
/aosp_15_r20/external/pytorch/benchmarks/distributed/ddp/
H A Dbenchmark.py162 def create_model(self): member in Benchmark
184 def create_model(self): member in TorchvisionBenchmark
/aosp_15_r20/external/executorch/examples/models/
H A Dmodel_factory.py20 def create_model( member in EagerModelFactory
/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
/aosp_15_r20/external/tensorflow/tensorflow/examples/speech_commands/
H A Dmodels.py91 def create_model(fingerprint_input, model_settings, model_architecture, function
/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
/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_padding.py199 def create_model(vocab_size): function
/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/python/typing/typing_extensions/src/
Dtest_typing_extensions.py2634 def create_model(*, frozen: bool = False, kw_only: bool = True): function
/aosp_15_r20/external/python/typing_extensions/src/
Dtest_typing_extensions.py5653 def create_model(*, frozen: bool = False, kw_only: bool = True): function
/aosp_15_r20/external/python/cpython3/Lib/test/
Dtest_typing.py8145 def create_model(*, frozen: bool = False, kw_only: bool = True): function