Home
last modified time | relevance | path

Searched refs:save_prefix (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/python/keras/preprocessing/
Dimage.py208 save_prefix='', argument
233 save_prefix=save_prefix,
279 save_prefix='', argument
299 save_prefix=save_prefix,
/external/tensorflow/tensorflow/python/keras/
Dmodels_test.py332 save_prefix = os.path.join(self.get_temp_dir(), 'ckpt')
335 model.save_weights(save_prefix)
337 model.load_weights(save_prefix)
/external/tensorflow/tensorflow/python/eager/
Dwrap_function_test.py122 save_prefix = os.path.join(self.get_temp_dir(), 'meta_graph_test')
131 saver.save(session, save_prefix)
134 saver_lib.import_meta_graph(save_prefix + '.meta')
/external/tensorflow/tensorflow/python/kernel_tests/
Dpartitioned_variables_test.py606 save_prefix = os.path.join(self.get_temp_dir(), "ckpt")
623 save_path = saver.save(sess=session, save_path=save_prefix)
Dlookup_ops_test.py1478 save_prefix = os.path.join(tempfile.mkdtemp(prefix=save_dir), "hash")
1502 save_path = save_checkpoint.save(save_prefix)
2867 save_prefix = os.path.join(tempfile.mkdtemp(prefix=save_dir), "hash")
2889 save_path = checkpoint.save(save_prefix)
/external/tensorflow/tensorflow/core/graph/
Dquantize_training.cc156 Node* FindRestoreAllOp(const Graph* graph, StringPiece save_prefix) { in FindRestoreAllOp() argument
159 if (node->name() == strings::StrCat(save_prefix, "/restore_all")) { in FindRestoreAllOp()
/external/tensorflow/tensorflow/python/keras/mixed_precision/experimental/
Dkeras_test.py1004 save_prefix = os.path.join(self.get_temp_dir(), 'ckpt')
1005 model.save_weights(save_prefix)
1013 model.load_weights(save_prefix)