Searched refs:SE_CUDA_DATA_HALF (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_blas.cc | 20 #define SE_CUDA_DATA_HALF CUDA_R_16F macro
346 static constexpr cudaDataType_t type = SE_CUDA_DATA_HALF;
1784 SE_CUDA_DATA_HALF, lda, GpuMemory(b), SE_CUDA_DATA_HALF, ldb, &beta, in DoBlasInternalImpl()
1785 GpuMemoryMutable(c), SE_CUDA_DATA_HALF, ldc); in DoBlasInternalImpl()
2643 a_matrix, SE_CUDA_DATA_HALF, lda, b_matrix, SE_CUDA_DATA_HALF, ldb, in DoBlasInternalImpl()
2644 &beta, c_matrix, SE_CUDA_DATA_HALF, ldc); in DoBlasInternalImpl()
|