Home
last modified time | relevance | path

Searched refs:VarToDataTypeMap (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/c/
Dcheckpoint_reader.cc59 var_to_data_type_map_.reset(new TensorSliceReader::VarToDataTypeMap( in CheckpointReader()
77 const TensorSliceReader::VarToDataTypeMap&
110 std::unique_ptr<TensorSliceReader::VarToDataTypeMap>>
136 std::unique_ptr<TensorSliceReader::VarToDataTypeMap> var_to_data_type_map( in BuildV2VarMaps()
137 new TensorSliceReader::VarToDataTypeMap); in BuildV2VarMaps()
Dcheckpoint_reader.h53 const TensorSliceReader::VarToDataTypeMap& GetVariableToDataTypeMap() const;
66 std::unique_ptr<TensorSliceReader::VarToDataTypeMap> >
74 std::unique_ptr<TensorSliceReader::VarToDataTypeMap> var_to_data_type_map_;
/external/tensorflow/tensorflow/core/util/
Dtensor_slice_reader.h105 typedef std::unordered_map<string, DataType> VarToDataTypeMap; typedef
111 VarToDataTypeMap GetVariableToDataTypeMap() const;
Dtensor_slice_reader.cc293 TensorSliceReader::VarToDataTypeMap
295 VarToDataTypeMap name_to_dtype; in GetVariableToDataTypeMap()