Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for BuildV2VarMaps (0.1 sec)

  1. 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)
  2. 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)
Back to top