TypedBCSRMatrix Class — pytorch Architecture

Source Code

aten/src/ATen/native/quantized/cpu/qnnpack/include/pack_block_sparse.h lines 107–125

template <typename INDICES_DTYPE>
struct TypedBCSRMatrix : BCSRMatrix {
  OwnedOrBorrowedVector<INDICES_DTYPE> col_indices;
  OwnedOrBorrowedVector<INDICES_DTYPE> row_values;
  TypedBCSRMatrix();
  const void* col_indices_data_ptr() const override;
  const void* row_values_data_ptr() const override;
#ifdef QNNPACK_BCSRMATRIX_DEBUG
  void print() const override;
#endif // QNNPACK_BCSRMATRIX_DEBUG
  void unpack(
      int8_t* dst,
      const int64_t num_rows,
      const int64_t num_cols,
      const uint8_t* zero_points) const override;
  uint32_t max_index() const override;

  ~TypedBCSRMatrix() override = default;
};