Searched defs:ckpt_reader (Results 1 – 10 of 10) sorted by relevance
42 explicit TFScope(checkpoint::CheckpointReader* ckpt_reader) in TFScope()
45 explicit TFGraph(checkpoint::CheckpointReader* ckpt_reader) in TFGraph()
46 std::unique_ptr<checkpoint::CheckpointReader> ckpt_reader( in TFProfTensorTest() local
41 explicit TFShow(checkpoint::CheckpointReader* ckpt_reader) in TFShow()
55 std::unique_ptr<checkpoint::CheckpointReader> ckpt_reader) in TFStats()78 std::unique_ptr<checkpoint::CheckpointReader> ckpt_reader) in TFStats()
68 std::unique_ptr<checkpoint::CheckpointReader> ckpt_reader( in TFProfShowTest() local
200 std::unique_ptr<checkpoint::CheckpointReader> ckpt_reader; in PrintModelAnalysis() local
62 std::unique_ptr<checkpoint::CheckpointReader> ckpt_reader( in TFProfStatsTest() local
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
189 std::unique_ptr<checkpoint::CheckpointReader> ckpt_reader; in Run() local