Home
last modified time | relevance | path

Searched refs:save_model (Results 1 – 14 of 14) sorted by relevance

/external/pytorch/test/mobile/lightweight_dispatch/
Dtests_setup.py16 def save_model(cls): function
36 @save_model
49 @save_model
60 @save_model
75 @save_model
87 @save_model
96 @save_model
105 @save_model
112 @save_model
121 @save_model
/external/pytorch/test/distributed/_composable/fully_shard/
Dtest_fully_shard_model_checkpoint.py137 save_model = copy.deepcopy(local_model)
139 save_model,
142 save_model.get_ignored_modules() if ignore_modules else []
148 FSDP.set_state_dict_type(save_model, StateDictType.FULL_STATE_DICT)
150 FSDP.set_state_dict_type(save_model, StateDictType.SHARDED_STATE_DICT)
151 state_dict = save_model.state_dict()
173 self._check_model_parity(load_model, save_model)
/external/tensorflow/tensorflow/python/keras/saving/
Dsave.py37 def save_model(model, function
/external/pytorch/torch/testing/_internal/
Dcommon_fsdp.py1204 save_model: bool = False,
1273 if save_model:
1291 save_model: bool = True,
1413 save_model=save_model,
/external/tensorflow/tensorflow/python/keras/
Dmodels.py44 save_model = save.save_model variable
/external/tensorflow/tensorflow/lite/python/
Dtflite_convert_test.py89 keras.models.save_model(model, keras_file)
100 keras.models.save_model(model, keras_file)
Dlite_test.py2275 keras.models.save_model(model, self._keras_file)
2436 keras.models.save_model(model, self._keras_file)
2497 keras.models.save_model(model, self._keras_file)
2641 keras.models.save_model(model, self._keras_file)
/external/pytorch/torch/onnx/_internal/
D_exporter_legacy.py823 onnx.save_model(self.model_proto, destination) # type: ignore[attr-defined]
827 onnx.save_model( # type: ignore[attr-defined]
/external/tensorflow/tensorflow/python/saved_model/
DREADME.md31 * [`tf.keras.models.save_model`](https://www.tensorflow.org/api_docs/python/tf/keras/models/save_mo…
/external/tensorflow/tensorflow/lite/g3doc/r1/convert/
Dpython_api.md87 tf.keras.models.save_model(model, keras_file)
/external/tensorflow/tensorflow/python/ops/linalg/
Dlinear_operator_test_util.py45 from tensorflow.python.saved_model import save as save_model unknown
877 save_model.save(m1, saved_model_dir)
/external/pytorch/benchmarks/dynamo/
Dcommon.py1638 onnx.save_model(model_proto, output_path, save_as_external_data=True)
1641 onnx.save_model(model_proto, output_path)
1669 onnx.save_model(
/external/tensorflow/tensorflow/python/keras/engine/
Dtraining.py2132 save.save_model(self, filepath, overwrite, include_optimizer, save_format,
/external/tensorflow/
DRELEASE.md4779 * Adds a `save_traces` argument to `model.save`/ `tf.keras.models.save_model`
6336 * `tf.keras.model.save_model` and `model.save` now defaults to saving a
6382 `tf.keras.models.save_model(..., save_format='tf')` and
6631 * `tf.keras.model.save_model` and `model.save` now defaults to saving a
6635 `tf.keras.models.save_model(..., save_format='tf')` and