Searched refs:v2_reader_ (Results 1 – 2 of 2) sorted by relevance
34 v2_reader_(nullptr), in CheckpointReader()42 v2_reader_.reset( in CheckpointReader()44 if (!v2_reader_->status().ok()) { in CheckpointReader()45 Set_TF_Status_from_Status(status, v2_reader_->status()); in CheckpointReader()68 return v2_reader_->Contains(name); in HasTensor()85 return v2_reader_->DebugString(); in DebugString()97 status = v2_reader_->LookupDtypeAndShape(name, &dtype, &shape); in GetTensor()100 status = v2_reader_->Lookup(name, out_tensor->get()); in GetTensor()112 CHECK(v2_reader_ != nullptr); in BuildV2VarMaps()113 CHECK(v2_reader_->status().ok()); in BuildV2VarMaps()[all …]
71 std::unique_ptr<BundleReader> v2_reader_; variable