Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for var_to_shape_map (0.07 sec)

  1. tensorflow/c/checkpoint_reader.cc

            << entry.InitializationErrorString();
        string key(v2_reader_->key());
        (*var_to_shape_map)[key] = TensorShape(entry.shape());
        (*var_to_data_type_map)[key] = DataType(entry.dtype());
      }
      // The returned pointers are owned by the caller.
      return std::make_pair(std::move(var_to_shape_map),
                            std::move(var_to_data_type_map));
    }
    
    }  // namespace checkpoint
    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

      // 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: 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