Searched defs:ckpt_reader (Results 1 – 10 of 10) sorted by relevance
156 const string& tensor_name, const std::unique_ptr<BundleReader>& ckpt_reader, in ReadTensorFromCheckpoint()183 std::unique_ptr<BundleReader>* ckpt_reader) { in InitializeCheckpointReader()227 const std::unique_ptr<BundleReader>& ckpt_reader, in SparsifyGatherInternal()596 std::unique_ptr<BundleReader> ckpt_reader; in SparsifyGather() local
54 std::unique_ptr<checkpoint::CheckpointReader> ckpt_reader) in TFStats()77 std::unique_ptr<checkpoint::CheckpointReader> ckpt_reader) in TFStats()
41 explicit TFScope(checkpoint::CheckpointReader* ckpt_reader) in TFScope()
43 explicit TFGraph(checkpoint::CheckpointReader* ckpt_reader) in TFGraph()
43 std::unique_ptr<checkpoint::CheckpointReader> ckpt_reader( in TFProfTensorTest() local
40 explicit TFShow(checkpoint::CheckpointReader* ckpt_reader) in TFShow()
65 std::unique_ptr<checkpoint::CheckpointReader> ckpt_reader( in TFProfShowTest() local
196 std::unique_ptr<checkpoint::CheckpointReader> ckpt_reader; in PrintModelAnalysis() local
58 std::unique_ptr<checkpoint::CheckpointReader> ckpt_reader( in TFProfStatsTest() local
184 std::unique_ptr<checkpoint::CheckpointReader> ckpt_reader; in Run() local