Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for TensorSliceReader (0.22 sec)

  1. tensorflow/c/checkpoint_reader.h

      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_;
    
      std::unique_ptr<TensorSliceReader::VarToShapeMap> var_to_shape_map_;
    C
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Thu Oct 12 08:49:52 GMT 2023
    - 3.1K bytes
    - Viewed (0)
  2. tensorflow/c/checkpoint_reader.cc

        }
      }
    
      // Second pass: adds the entries, ignoring the filtered keys.
      std::unique_ptr<TensorSliceReader::VarToShapeMap> var_to_shape_map(
          new TensorSliceReader::VarToShapeMap);
      std::unique_ptr<TensorSliceReader::VarToDataTypeMap> var_to_data_type_map(
          new TensorSliceReader::VarToDataTypeMap);
      v2_reader_->Seek(kHeaderEntryKey);
      for (v2_reader_->Next(); v2_reader_->Valid(); v2_reader_->Next()) {
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Fri Aug 25 21:29:12 GMT 2023
    - 5.5K bytes
    - Viewed (0)
Back to top