- Sort Score
- Num 10 results
- Language All
Results 1 - 2 of 2 for var_to_shape_map (0.09 seconds)
-
tensorflow/c/checkpoint_reader.cc
<< entry.InitializationErrorString(); string key(v2_reader_->key()); (*var_to_shape_map)[key] = TensorShape(entry.shape()); (*var_to_data_type_map)[key] = DataType(entry.dtype()); } // The returned pointers are owned by the caller. return std::make_pair(std::move(var_to_shape_map), std::move(var_to_data_type_map)); } } // namespace checkpoint
Created: Tue Dec 30 12:39:10 GMT 2025 - Last Modified: Sat Nov 08 06:24:11 GMT 2025 - 5.4K bytes - Click Count (0) -
tensorflow/c/checkpoint_reader.h
// Invariant: exactly one of "reader_" and "v2_reader_" is non-null. std::unique_ptr<TensorSliceReader> reader_; std::unique_ptr<BundleReader> v2_reader_; std::unique_ptr<TensorSliceReader::VarToShapeMap> var_to_shape_map_; std::unique_ptr<TensorSliceReader::VarToDataTypeMap> var_to_data_type_map_; CheckpointReader(const CheckpointReader&) = delete; void operator=(const CheckpointReader&) = delete; };
Created: Tue Dec 30 12:39:10 GMT 2025 - Last Modified: Thu Oct 12 08:49:52 GMT 2023 - 3.1K bytes - Click Count (0)