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
42 explicit TFShow(checkpoint::CheckpointReader* ckpt_reader) in TFShow()
52 std::unique_ptr<checkpoint::CheckpointReader> ckpt_reader) in TFStats()76 std::unique_ptr<checkpoint::CheckpointReader> ckpt_reader) in TFStats()
68 std::unique_ptr<checkpoint::CheckpointReader> ckpt_reader( in TFProfShowTest() local
195 std::unique_ptr<checkpoint::CheckpointReader> ckpt_reader; in PrintModelAnalysis() local
62 std::unique_ptr<checkpoint::CheckpointReader> ckpt_reader( in TFProfStatsTest() local
157 const string& tensor_name, const std::unique_ptr<BundleReader>& ckpt_reader, in ReadTensorFromCheckpoint()184 std::unique_ptr<BundleReader>* ckpt_reader) { in InitializeCheckpointReader()228 const std::unique_ptr<BundleReader>& ckpt_reader, in SparsifyGatherInternal()597 std::unique_ptr<BundleReader> ckpt_reader; in SparsifyGather() local
187 std::unique_ptr<checkpoint::CheckpointReader> ckpt_reader; in Run() local