pytorch Architecture Docs
Home / Class/ convert_int_to_enum Class — pytorch Architecture

convert_int_to_enum Class — pytorch Architecture

Architecture documentation for the convert_int_to_enum class in CUDAScaledBlas.h from the pytorch codebase.

Class c

Entity Profile

Source Code

aten/src/ATen/cuda/CUDAScaledBlas.h lines 109–118

template <class EnumType, class ArrayType>
std::vector<EnumType> convert_int_to_enum(ArrayType& v) {
  std::vector<EnumType> converted;
  converted.reserve(v.size());

  for (auto vi : v) {
    converted.push_back(static_cast<EnumType>(vi));
  }
  return converted;
}

Source

Analyze Your Own Codebase

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

Try Supermodel Free