Searched refs:ROCMBlasTranspose (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_blas.cc | 333 rocblas_operation ROCMBlasTranspose(blas::Transpose trans) { in ROCMBlasTranspose() function
879 ROCMBlasTranspose(trans), m, n, &alpha, GpuMemory(a), lda, GpuMemory(x), in DoBlasGemv()
889 ROCMBlasTranspose(trans), m, n, &alpha, GpuMemory(a), lda, GpuMemory(x), in DoBlasGemv()
1497 ROCMBlasTranspose(transa), ROCMBlasTranspose(transb), m, n, k, in DoBlasGemm()
1540 ROCMBlasTranspose(transa), ROCMBlasTranspose(transb), m, n, k, &alpha, in DoBlasGemm()
1551 ROCMBlasTranspose(transa), ROCMBlasTranspose(transb), m, n, k, &alpha, in DoBlasGemm()
1875 if (ROCMBlasTranspose(transa) == rocblas_operation_none) in DoBlasGemmBatchedInternal()
1880 if (ROCMBlasTranspose(transb) == rocblas_operation_none) in DoBlasGemmBatchedInternal()
1891 ROCMBlasTranspose(transa), ROCMBlasTranspose(transb), m, n, k, in DoBlasGemmBatchedInternal()
2234 ROCMBlasSide(side), ROCMBlasUpperLower(uplo), ROCMBlasTranspose(transa), in DoBlasTrsm()
[all …]
|