Home
last modified time | relevance | path

Searched refs:v2_reader_ (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/c/
Dcheckpoint_reader.cc35 v2_reader_(nullptr), in CheckpointReader()
43 v2_reader_.reset( in CheckpointReader()
45 if (!v2_reader_->status().ok()) { in CheckpointReader()
46 Set_TF_Status_from_Status(out_status, v2_reader_->status()); in CheckpointReader()
69 return v2_reader_->Contains(name); in HasTensor()
86 return v2_reader_->DebugString(); in DebugString()
98 status = v2_reader_->LookupDtypeAndShape(name, &dtype, &shape); in GetTensor()
101 status = v2_reader_->Lookup(name, out_tensor->get()); in GetTensor()
113 CHECK(v2_reader_ != nullptr); in BuildV2VarMaps()
114 CHECK(v2_reader_->status().ok()); in BuildV2VarMaps()
[all …]
Dcheckpoint_reader.h71 std::unique_ptr<BundleReader> v2_reader_; variable