- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for BuildV2VarMaps (0.1 sec)
-
tensorflow/c/checkpoint_reader.cc
new BundleReader(Env::Default(), filename /* prefix to a V2 ckpt */)); if (!v2_reader_->status().ok()) { tsl::Set_TF_Status_from_Status(status, v2_reader_->status()); return; } auto result = BuildV2VarMaps(); var_to_shape_map_.swap(result.first); var_to_data_type_map_.swap(result.second); } else { reader_.reset(new TensorSliceReader(filename)); if (!reader_->status().ok()) {
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 16:27:48 UTC 2024 - 5.6K bytes - Viewed (0) -
tensorflow/c/checkpoint_reader.h
// REQUIRES: "v2_reader_ != nullptr && v2_reader_.status().ok()". std::pair<std::unique_ptr<TensorSliceReader::VarToShapeMap>, std::unique_ptr<TensorSliceReader::VarToDataTypeMap> > BuildV2VarMaps(); // Invariant: exactly one of "reader_" and "v2_reader_" is non-null. std::unique_ptr<TensorSliceReader> reader_; std::unique_ptr<BundleReader> v2_reader_;
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Thu Oct 12 08:49:52 UTC 2023 - 3.1K bytes - Viewed (0)