Searched refs:VarToShapeMap (Results 1 – 4 of 4) sorted by relevance
49 const TensorSliceReader::VarToShapeMap& GetVariableToShapeMap() const;65 std::pair<std::unique_ptr<TensorSliceReader::VarToShapeMap>,73 std::unique_ptr<TensorSliceReader::VarToShapeMap> var_to_shape_map_;
58 new TensorSliceReader::VarToShapeMap(reader_->GetVariableToShapeMap())); in CheckpointReader()71 const TensorSliceReader::VarToShapeMap&109 std::pair<std::unique_ptr<TensorSliceReader::VarToShapeMap>,134 std::unique_ptr<TensorSliceReader::VarToShapeMap> var_to_shape_map( in BuildV2VarMaps()135 new TensorSliceReader::VarToShapeMap); in BuildV2VarMaps()
104 typedef std::unordered_map<string, TensorShape> VarToShapeMap; typedef108 VarToShapeMap GetVariableToShapeMap() const;
282 TensorSliceReader::VarToShapeMap TensorSliceReader::GetVariableToShapeMap() in GetVariableToShapeMap()284 VarToShapeMap name_to_shape; in GetVariableToShapeMap()