Search Options

Display Count
Sort
Preferred Language
Advanced Search

Results 1 - 1 of 1 for LookupDtypeAndShape (0.17 seconds)

  1. 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();
        }
    Created: Tue Apr 07 12:39:13 GMT 2026
    - Last Modified: Tue Feb 17 18:49:14 GMT 2026
    - 5.4K bytes
    - Click Count (0)
Back to Top