gemm Class — pytorch Architecture

Architecture documentation for the gemm class in CUDABlas.h from the pytorch codebase.

Class c

Entity Profile

Source Code

aten/src/ATen/cuda/CUDABlas.h lines 60–61

template <typename Dtype, typename C_Dtype, typename std::enable_if<CUDABLAS_GEMM_DTYPE_IS_FLOAT_TYPE_AND_C_DTYPE_IS_FLOAT, Dtype>::type* = nullptr>
void gemm(CUDABLAS_GEMM_ARGTYPES_AND_C_DTYPE(Dtype, C_Dtype));

Source

View on GitHub

Analyze Your Own Codebase

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

Try Supermodel Free