Searched refs:VarToDataTypeMap (Results 1 – 5 of 5) sorted by relevance
60 var_to_data_type_map_.reset(new TensorSliceReader::VarToDataTypeMap( in CheckpointReader()78 const TensorSliceReader::VarToDataTypeMap&111 std::unique_ptr<TensorSliceReader::VarToDataTypeMap>>137 std::unique_ptr<TensorSliceReader::VarToDataTypeMap> var_to_data_type_map( in BuildV2VarMaps()138 new TensorSliceReader::VarToDataTypeMap); in BuildV2VarMaps()
53 const TensorSliceReader::VarToDataTypeMap& GetVariableToDataTypeMap() const;66 std::unique_ptr<TensorSliceReader::VarToDataTypeMap> >74 std::unique_ptr<TensorSliceReader::VarToDataTypeMap> var_to_data_type_map_;
105 typedef std::unordered_map<string, DataType> VarToDataTypeMap; typedef111 VarToDataTypeMap GetVariableToDataTypeMap() const;
288 TensorSliceReader::VarToDataTypeMap290 VarToDataTypeMap name_to_dtype; in GetVariableToDataTypeMap()
71 %typemap(out) const tensorflow::checkpoint::TensorSliceReader::VarToDataTypeMap& {