Searched defs:Adesc (Results 1 – 1 of 1) sorted by relevance
| /external/pytorch/aten/src/ATen/cuda/ |
| D | CUDABlas.cpp | 361 CuBlasLtMatrixLayout Adesc(abcType, m, k, lda, opa == CUBLAS_OP_T); in bgemm_internal_cublaslt() local
1239 CuBlasLtMatrixLayout Adesc(abcType, m, k, mat1_ld, transpose_mat1); in gemm_and_bias() local
1434 CuBlasLtMatrixLayout Adesc(ScalarTypeToCudaDataType(mat1_dtype), m, k, mat1_ld, transa == 't'); in scaled_gemm() local
1593 CuBlasLtMatrixLayout Adesc(abType, m, k, mat1_ld, transpose_mat1); in int8_gemm() local
|