Home
last modified time | relevance | path

Searched refs:GetVariableToShapeMap (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/c/
Dcheckpoint_reader.cc59 new TensorSliceReader::VarToShapeMap(reader_->GetVariableToShapeMap())); in CheckpointReader()
73 CheckpointReader::GetVariableToShapeMap() const { in GetVariableToShapeMap() function in tensorflow::checkpoint::CheckpointReader
Dcheckpoint_reader.h49 const TensorSliceReader::VarToShapeMap& GetVariableToShapeMap() const;
/external/tensorflow/tensorflow/core/util/
Dtensor_slice_reader.h108 VarToShapeMap GetVariableToShapeMap() const;
Dtensor_slice_reader.cc277 TensorSliceReader::VarToShapeMap TensorSliceReader::GetVariableToShapeMap() in GetVariableToShapeMap() function in tensorflow::checkpoint::TensorSliceReader
/external/tensorflow/tensorflow/python/util/
Dpy_checkpoint_reader.i136 …ename("get_variable_to_shape_map") tensorflow::checkpoint::CheckpointReader::GetVariableToShapeMap;
/external/tensorflow/tensorflow/core/profiler/internal/
Dtfprof_stats.cc66 for (const auto& v : ckpt_reader_->GetVariableToShapeMap()) { in TFStats()