- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for LookupDtypeAndShape (2.74 sec)
-
tensorflow/c/checkpoint_reader.cc
absl::Status status; if (reader_ != nullptr) { status = reader_->GetTensor(name, out_tensor); } else { tensorflow::DataType dtype; tensorflow::TensorShape shape; status = v2_reader_->LookupDtypeAndShape(name, &dtype, &shape); if (status.ok()) { out_tensor->reset(new Tensor(dtype, shape)); status = v2_reader_->Lookup(name, out_tensor->get()); if (!status.ok()) out_tensor->reset(); }
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 16:27:48 UTC 2024 - 5.6K bytes - Viewed (0)