Searched defs:Cdesc (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/cuda/ |
| H A D | CUDABlas.cpp | 363 CuBlasLtMatrixLayout Cdesc(abcType, m, n, ldc); in bgemm_internal_cublaslt() local
1241 CuBlasLtMatrixLayout Cdesc(abcType, m, n, result_ld); in gemm_and_bias() local
1438 CuBlasLtMatrixLayout Cdesc(ScalarTypeToCudaDataType(result_dtype), m, n, result_ld); in scaled_gemm() local
1440 CuBlasLtMatrixLayout Cdesc(ScalarTypeToCudaDataType(bias_dtype), m, n, result_ld); in scaled_gemm() local
1595 CuBlasLtMatrixLayout Cdesc(cType, m, n, result_ld); in int8_gemm() local
|