Searched refs:GetVariableToDataTypeMap (Results 1 – 10 of 10) sorted by relevance
119 auto data_type = reader.GetVariableToDataTypeMap().find(entry.name()); in DEFINE_PROTO_FUZZER()121 if (data_type != reader.GetVariableToDataTypeMap().end() && in DEFINE_PROTO_FUZZER()
60 reader_->GetVariableToDataTypeMap())); in CheckpointReader()78 CheckpointReader::GetVariableToDataTypeMap() const { in GetVariableToDataTypeMap() function in tensorflow::checkpoint::CheckpointReader
53 const TensorSliceReader::VarToDataTypeMap& GetVariableToDataTypeMap() const;
357 const auto& m = reader->GetVariableToDataTypeMap(); in TF_NewCheckpointReader()382 const auto& m = reader->GetVariableToDataTypeMap(); in TF_CheckpointReaderGetVariableDataType()
56 tf_checkpoint_reader->GetVariableToDataTypeMap()) { in Create()
146 &tensorflow::checkpoint::CheckpointReader::GetVariableToDataTypeMap) in PYBIND11_MODULE()
111 VarToDataTypeMap GetVariableToDataTypeMap() const;
301 TensorSliceReader::GetVariableToDataTypeMap() const { in GetVariableToDataTypeMap() function in tensorflow::checkpoint::TensorSliceReader
162 tf_checkpoint_reader.GetVariableToDataTypeMap()) { in ReadTensors()
133 tensorflow::checkpoint::CheckpointReader::GetVariableToDataTypeMap