Home / Class/ max_index Class — pytorch Architecture

max_index Class — pytorch Architecture

Architecture documentation for the max_index class in pack_block_sparse.h from the pytorch codebase.

Entity Profile

Source Code

aten/src/ATen/native/quantized/cpu/qnnpack/include/pack_block_sparse.h lines 332–339

template <typename INDICES_DTYPE>
uint32_t TypedBCSRMatrix<INDICES_DTYPE>::max_index() const {
  return static_cast<uint32_t>(std::max(
      *std::max_element(
          row_values.data(), row_values.data() + row_values.size()),
      *std::max_element(
          col_indices.data(), col_indices.data() + col_indices.size())));
}

Analyze Your Own Codebase

Get architecture documentation, dependency graphs, and domain analysis for your codebase in minutes.

Try Supermodel Free