Searched refs:SE_CUDA_DATA_HALF (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_blas.cc | 19 #define SE_CUDA_DATA_HALF CUDA_R_16F macro
330 static constexpr cudaDataType_t type = SE_CUDA_DATA_HALF;
1612 SE_CUDA_DATA_HALF, lda, GpuMemory(b), SE_CUDA_DATA_HALF, ldb, &beta, in DoBlasGemm()
1613 GpuMemoryMutable(c), SE_CUDA_DATA_HALF, ldc); in DoBlasGemm()
2404 CUDABlasTranspose(transb), m, n, k, &alpha, a_matrix, SE_CUDA_DATA_HALF, in DoBlasGemmStridedBatched()
2405 lda, b_matrix, SE_CUDA_DATA_HALF, ldb, &beta, c_matrix, in DoBlasGemmStridedBatched()
2406 SE_CUDA_DATA_HALF, ldc); in DoBlasGemmStridedBatched()
|