Searched refs:GetVariableToShapeMap (Results 1 – 11 of 11) sorted by relevance
120 auto shape = reader.GetVariableToShapeMap().find(entry.name()); in DEFINE_PROTO_FUZZER()122 shape != reader.GetVariableToShapeMap().end() && in DEFINE_PROTO_FUZZER()
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;
397 const auto& shape = reader->GetVariableToShapeMap().at(name); in TF_CheckpointReaderGetVariableShape()411 const auto& m = reader->GetVariableToShapeMap(); in TF_CheckpointReaderGetVariableNumDims()
64 tf_checkpoint_reader->GetVariableToShapeMap()) { in Create()
86 for (const auto& [name, shape] : reader->GetVariableToShapeMap()) { in SummarizeCheckpoint()
144 &tensorflow::checkpoint::CheckpointReader::GetVariableToShapeMap) in PYBIND11_MODULE()
108 VarToShapeMap GetVariableToShapeMap() const;
289 TensorSliceReader::VarToShapeMap TensorSliceReader::GetVariableToShapeMap() in GetVariableToShapeMap() function in tensorflow::checkpoint::TensorSliceReader
67 for (const auto& v : ckpt_reader_->GetVariableToShapeMap()) { in TFStats()
132 tensorflow::checkpoint::CheckpointReader::GetVariableToShapeMap