Searched refs:CUDABlasDiagonal (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_blas.cc | 313 cublasDiagType_t CUDABlasDiagonal(blas::Diagonal diag) { in CUDABlasDiagonal() function
1500 CUDABlasDiagonal(diag), n, k, GpuMemory(a), lda, in DoBlasInternalImpl()
1510 CUDABlasDiagonal(diag), n, k, GpuMemory(a), lda, in DoBlasInternalImpl()
1521 CUDABlasDiagonal(diag), n, k, GpuComplex(GpuMemory(a)), in DoBlasInternalImpl()
1532 CUDABlasDiagonal(diag), n, k, GpuComplex(GpuMemory(a)), in DoBlasInternalImpl()
1542 CUDABlasDiagonal(diag), n, k, GpuMemory(a), lda, in DoBlasInternalImpl()
1552 CUDABlasDiagonal(diag), n, k, GpuMemory(a), lda, in DoBlasInternalImpl()
1563 CUDABlasDiagonal(diag), n, k, GpuComplex(GpuMemory(a)), in DoBlasInternalImpl()
1574 CUDABlasDiagonal(diag), n, k, GpuComplex(GpuMemory(a)), in DoBlasInternalImpl()
1584 CUDABlasDiagonal(diag), n, GpuMemory(ap), in DoBlasInternalImpl()
[all …]
|