Searched refs:CreateCublasLtLayoutDesc (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_blas.cc | 3206 port::StatusOr<UniqueLayoutDesc> CreateCublasLtLayoutDesc( in DoBlasInternalImpl() function
3273 a_desc_, CreateCublasLtLayoutDesc(p.ab_type, rows_a, cols_a, p.lda, in DoBlasInternalImpl()
3276 b_desc_, CreateCublasLtLayoutDesc(p.ab_type, rows_b, cols_b, p.ldb, in DoBlasInternalImpl()
3279 c_desc_, CreateCublasLtLayoutDesc(p.c_type, p.m, p.n, p.ldc, p.stride_c, in DoBlasInternalImpl()
3282 d_desc_, CreateCublasLtLayoutDesc(p.c_type, p.m, p.n, p.ldc, p.stride_c, in DoBlasInternalImpl()
3289 CreateCublasLtLayoutDesc(p.ab_type, rows_a, cols_a, p.lda, p.stride_a, in DoBlasInternalImpl()
3293 CreateCublasLtLayoutDesc(p.ab_type, rows_b, cols_b, p.ldb, p.stride_b, in DoBlasInternalImpl()
3297 CreateCublasLtLayoutDesc(p.c_type, p.m, p.n, p.ldc, p.stride_c, in DoBlasInternalImpl()
3301 CreateCublasLtLayoutDesc(p.c_type, p.m, p.n, p.ldc, p.stride_c, in DoBlasInternalImpl()
|