print Class — pytorch Architecture

Source Code

aten/src/ATen/native/quantized/cpu/qnnpack/include/pack_block_sparse.h lines 264–289

template <typename INDICES_DTYPE>
void TypedBCSRMatrix<INDICES_DTYPE>::print() const {
  std::cout << "row block size:" << row_block_size << std::endl;
  std::cout << "col block size:" << col_block_size << std::endl;
  std::cout << "row ptr\n";
  std::cout
      << "indices dtype: uint"
      << static_cast<
             std::underlying_type_t<pytorch_qnnp_sparse_matrix_indices_dtype>>(
             indices_dtype)
      << "_t" << std::endl;
  for (uint32_t i = 0; i < row_values.size(); i++) {
    std::cout << (uint32_t)row_values[i] << ", ";
  }
  std::cout << std::endl;
  std::cout << "col indices\n";
  for (uint32_t i = 0; i < col_indices.size(); i++) {
    std::cout << (uint32_t)col_indices[i] << ", ";
  }
  std::cout << std::endl;
  std::cout << "Actual values\n";
  for (uint32_t i = 0; i < values.size(); i++) {
    std::cout << (uint32_t)values[i] << ", ";
  }
  std::cout << std::endl;
}