Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for var_to_shape_map_ (0.4 sec)

  1. tensorflow/c/checkpoint_reader.cc

          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()) {
          tsl::Set_TF_Status_from_Status(status, reader_->status());
          return;
        }
        var_to_shape_map_.reset(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Aug 25 21:29:12 UTC 2023
    - 5.5K bytes
    - Viewed (0)
  2. 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;
    };
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Oct 12 08:49:52 UTC 2023
    - 3.1K bytes
    - Viewed (0)
Back to top