Searched refs:ckpt_reader_ (Results 1 – 4 of 4) sorted by relevance
55 ckpt_reader_(std::move(ckpt_reader)) { in TFStats()65 if (ckpt_reader_) { in TFStats()66 for (const auto& v : ckpt_reader_->GetVariableToShapeMap()) { in TFStats()79 ckpt_reader_(std::move(ckpt_reader)) { in TFStats()109 scope_view_.reset(new TFScope(ckpt_reader_.get())); in BuildView()116 graph_view_.reset(new TFGraph(ckpt_reader_.get())); in BuildView()
43 : ckpt_reader_(ckpt_reader) {} in TFShow()121 checkpoint::CheckpointReader* ckpt_reader_; variable
112 std::unique_ptr<checkpoint::CheckpointReader> ckpt_reader_; variable
55 if (name == kTFProfRoot || !ckpt_reader_ || !tensor) { in LookUpCheckPoint()60 ckpt_reader_->GetTensor(name, &out_tensor, status); in LookUpCheckPoint()