Searched refs:GetVariableToShapeMap (Results 1 – 8 of 8) sorted by relevance
58 new TensorSliceReader::VarToShapeMap(reader_->GetVariableToShapeMap())); in CheckpointReader()72 CheckpointReader::GetVariableToShapeMap() const { in GetVariableToShapeMap() function in tensorflow::checkpoint::CheckpointReader
49 const TensorSliceReader::VarToShapeMap& GetVariableToShapeMap() const;
388 const auto& shape = reader->GetVariableToShapeMap().at(name); in TF_CheckpointReaderGetVariableShape()402 const auto& m = reader->GetVariableToShapeMap(); in TF_CheckpointReaderGetVariableNumDims()
144 &tensorflow::checkpoint::CheckpointReader::GetVariableToShapeMap) in PYBIND11_MODULE()
108 VarToShapeMap GetVariableToShapeMap() const;
282 TensorSliceReader::VarToShapeMap TensorSliceReader::GetVariableToShapeMap() in GetVariableToShapeMap() function in tensorflow::checkpoint::TensorSliceReader
68 for (const auto& v : ckpt_reader_->GetVariableToShapeMap()) { in TFStats()
123 tensorflow::checkpoint::CheckpointReader::GetVariableToShapeMap