- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for GetVariableToShapeMap (0.17 sec)
-
tensorflow/c/checkpoint_reader.cc
if (!reader_->status().ok()) { tsl::Set_TF_Status_from_Status(status, reader_->status()); return; } var_to_shape_map_.reset( new TensorSliceReader::VarToShapeMap(reader_->GetVariableToShapeMap())); var_to_data_type_map_.reset(new TensorSliceReader::VarToDataTypeMap( reader_->GetVariableToDataTypeMap())); } } bool CheckpointReader::HasTensor(const string& name) const {
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Aug 25 21:29:12 GMT 2023 - 5.5K bytes - Viewed (0) -
tensorflow/c/checkpoint_reader.h
const string DebugString() const; // Returns a map from variable names to their shapes. Slices of a partitioned // tensor are combined into a single entry. const TensorSliceReader::VarToShapeMap& GetVariableToShapeMap() const; // Returns a map from variable names to their data types. Slices of a // partitioned tensor are combined into a single entry. const TensorSliceReader::VarToDataTypeMap& GetVariableToDataTypeMap() const;
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Oct 12 08:49:52 GMT 2023 - 3.1K bytes - Viewed (0)