Searched refs:VarToShapeMap (Results 1 – 5 of 5) sorted by relevance
59 new TensorSliceReader::VarToShapeMap(reader_->GetVariableToShapeMap())); in CheckpointReader()72 const TensorSliceReader::VarToShapeMap&110 std::pair<std::unique_ptr<TensorSliceReader::VarToShapeMap>,135 std::unique_ptr<TensorSliceReader::VarToShapeMap> var_to_shape_map( in BuildV2VarMaps()136 new TensorSliceReader::VarToShapeMap); in BuildV2VarMaps()
49 const TensorSliceReader::VarToShapeMap& GetVariableToShapeMap() const;65 std::pair<std::unique_ptr<TensorSliceReader::VarToShapeMap>,73 std::unique_ptr<TensorSliceReader::VarToShapeMap> var_to_shape_map_;
104 typedef std::unordered_map<string, TensorShape> VarToShapeMap; typedef108 VarToShapeMap GetVariableToShapeMap() const;
277 TensorSliceReader::VarToShapeMap TensorSliceReader::GetVariableToShapeMap() in GetVariableToShapeMap()279 VarToShapeMap name_to_shape; in GetVariableToShapeMap()
27 %typemap(out) const tensorflow::checkpoint::TensorSliceReader::VarToShapeMap& {