Searched refs:checkpoint_id (Results 1 – 3 of 3) sorted by relevance
/external/tensorflow/tensorflow/python/training/tracking/ |
D | graph_view.py | 245 for checkpoint_id, (trackable, object_proto) in enumerate( 247 assert node_ids[trackable] == checkpoint_id 348 for checkpoint_id, trackable in enumerate(trackable_objects): 349 assert node_ids[trackable] == checkpoint_id
|
/external/tensorflow/tensorflow/core/kernels/data/experimental/ |
D | snapshot_util.h | 70 const uint64 checkpoint_id); 323 const std::string& shard_directory, uint64 checkpoint_id, 340 uint64 checkpoint_id, const std::string& compression,
|
D | snapshot_util.cc | 75 uint64 checkpoint_id) { in GetCheckpointFileName() argument 79 static_cast<unsigned long long>(checkpoint_id))); in GetCheckpointFileName() 940 uint64 checkpoint_id, const std::string& compression, in AsyncWriter() argument 945 [this, env, shard_directory, checkpoint_id, compression, version, in AsyncWriter() 947 done(WriterThread(env, shard_directory, checkpoint_id, compression, in AsyncWriter() 976 uint64 checkpoint_id, in WriterThread() argument 983 env, GetCheckpointFileName(shard_directory, checkpoint_id), compression, in WriterThread()
|