Searched refs:ROCMBlasDiagonal (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_blas.cc | 357 rocblas_diagonal ROCMBlasDiagonal(blas::Diagonal diag) { in ROCMBlasDiagonal() function
2235 ROCMBlasDiagonal(diag), m, n, &alpha, const_cast<float *>(GpuMemory(a)), in DoBlasTrsm()
2247 ROCMBlasDiagonal(diag), m, n, &alpha, const_cast<double *>(GpuMemory(a)), in DoBlasTrsm()
|