Home
last modified time | relevance | path

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

/external/pytorch/test/distributed/_composable/
Dtest_compose.py57 base_model: nn.Module,
65 base_optim = torch.optim.Adam(base_model.parameters(), lr=LR)
73 base_loss = base_model(x).sum()
89 base_model = copy.deepcopy(model)
101 "base_model": [base_model],
114 base_model = copy.deepcopy(model)
125 "base_model": [base_model],
209 base_model = copy.deepcopy(model)
218 "base_model": [base_model],
232 base_model = copy.deepcopy(model)
[all …]
/external/pytorch/test/distributed/checkpoint/fsdp/
Dtest_fsdp_dsd.py37 base_model = nn.Sequential(
42 return base_model
55 base_model = self._get_base_model(mlp_dim)
57 model1 = copy.deepcopy(base_model)
68 model2 = copy.deepcopy(base_model)
200 base_model = nn.Sequential(MLP(mlp_dim), MLP(mlp_dim), MLP(mlp_dim))
201 return base_model
213 base_model = _get_base_model().cuda()
214 base_optim = torch.optim.AdamW(base_model.parameters(), lr=0.1)
218 copy.deepcopy(base_model),
[all …]
/external/pytorch/test/onnx/
Dtest_models_quantized_onnxruntime.py35 def __init__(self, base_model): argument
37 self.base_model = base_model
40 x = self.base_model(x)
/external/vboot_reference/futility/
Dupdater_manifest.c375 struct model_config *base_model; in manifest_from_signer_config() local
389 base_model = manifest_get_model_config(manifest, base_name); in manifest_from_signer_config()
391 if (!base_model) { in manifest_from_signer_config()
393 } else if (!base_model->has_custom_label) { in manifest_from_signer_config()
394 base_model->has_custom_label = true; in manifest_from_signer_config()
609 const struct model_config *base_model) in manifest_find_custom_label_model() argument
618 model = quirk_override_custom_label(cfg, manifest, base_model); in manifest_find_custom_label_model()
635 VPD_CUSTOM_LABEL_TAG, base_model->name); in manifest_find_custom_label_model()
636 return base_model; in manifest_find_custom_label_model()
639 VB2_DEBUG("Found custom label tag: %s (base=%s)\n", tag, base_model->name); in manifest_find_custom_label_model()
[all …]
Dupdater.h388 const struct model_config *base_model);
Dupdater.c1439 const struct model_config *base_model = model; in updater_setup_archive() local
1440 model = manifest_find_custom_label_model(cfg, manifest, base_model); in updater_setup_archive()
1450 if (base_model->image) { in updater_setup_archive()
1452 strcmp(base_model->image, model->image)) { in updater_setup_archive()
1455 base_model->name, model->name); in updater_setup_archive()
/external/cpuinfo/src/x86/
Dinfo.c9 model_info.base_model = (eax >> 4) & 0xF; in cpuinfo_x86_decode_model_info()
16 model_info.model = model_info.base_model + (model_info.extended_model << 4); in cpuinfo_x86_decode_model_info()
Dapi.h40 uint32_t base_model; member
/external/pytorch/test/distributed/_composable/fully_shard/
Dtest_fully_shard_compile.py75 base_model = TransformerWithSharedParams.init(
81 ref_model = fully_shard(copy.deepcopy(base_model), **fsdp_kwargs)
83 model = fully_shard(copy.deepcopy(base_model), **fsdp_kwargs)
/external/rust/android-crates-io/crates/quiche/deps/boringssl/src/crypto/
Dcpu-intel.c180 const uint32_t base_model = (eax >> 4) & 15; in OPENSSL_cpuid_setup() local
183 uint32_t model = base_model; in OPENSSL_cpuid_setup()
/external/cronet/tot/third_party/boringssl/src/crypto/
Dcpu_intel.cc176 const uint32_t base_model = (eax >> 4) & 15; in OPENSSL_cpuid_setup() local
179 uint32_t model = base_model; in OPENSSL_cpuid_setup()
/external/cronet/stable/third_party/boringssl/src/crypto/
Dcpu_intel.cc176 const uint32_t base_model = (eax >> 4) & 15; in OPENSSL_cpuid_setup() local
179 uint32_t model = base_model; in OPENSSL_cpuid_setup()
/external/boringssl/src/crypto/
Dcpu_intel.cc145 const uint32_t base_model = (eax >> 4) & 15; in OPENSSL_cpuid_setup() local
148 uint32_t model = base_model; in OPENSSL_cpuid_setup()
/external/pytorch/test/distributed/_composable/fsdp/
Dtest_fully_shard_state_dict.py44 base_model = nn.Sequential(
50 model1 = copy.deepcopy(base_model)
57 model2 = copy.deepcopy(base_model)
/external/pytorch/docs/source/
Dfunc.migrating.rst187 base_model = copy.deepcopy(models[0])
188 base_model.to('meta')
195 return torch.func.functional_call(base_model, (params, buffers), (data,))
/external/google-cloud-java/java-automl/proto-google-cloud-automl-v1beta1/src/main/proto/google/cloud/automl/v1beta1/
Dtranslation.proto53 string base_model = 1; field
/external/googleapis/google/cloud/automl/v1/
Dtranslation.proto54 string base_model = 1; field
/external/google-cloud-java/java-automl/proto-google-cloud-automl-v1/src/main/proto/google/cloud/automl/v1/
Dtranslation.proto54 string base_model = 1; field
/external/googleapis/google/cloud/automl/v1beta1/
Dtranslation.proto53 string base_model = 1; field
/external/googleapis/google/ai/generativelanguage/v1beta3/
Dtuned_model.proto61 string base_model = 4 [ field
143 string base_model = 2 [ field
/external/googleapis/google/ai/generativelanguage/v1beta/
Dtuned_model.proto61 string base_model = 4 [ field
143 string base_model = 2 [ field
/external/pytorch/test/distributed/fsdp/
Dtest_fsdp_use_orig_params.py252 base_model = TransformerWithSharedParams.init(
258 ref_model = FSDP(copy.deepcopy(base_model), self.process_group, **fsdp_kwargs)
260 model = FSDP(copy.deepcopy(base_model), self.process_group, **fsdp_kwargs)
/external/pytorch/benchmarks/functional_autograd_benchmark/
Dtorchvision_models.py446 base_model = model_map[name][1]
448 model = base_model(backbone, classifier, aux_classifier)
/external/googleapis/google/cloud/aiplatform/v1/
Dtuning_job.proto45 string base_model = 4; field
/external/pytorch/torch/testing/_internal/distributed/
Ddistributed_test.py8729 base_model = self._test_different_graph_across_ranks(
8733 base_model._get_ddp_logging_data().get("has_rebuilt_buckets", 0)
8739 for i, j in zip(base_model.parameters(), static_model.parameters()):