Searched refs:alpha_half (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_blas.cc | 1493 const Eigen::half alpha_half(alpha); in DoBlasGemm() local
1498 reinterpret_cast<const rocblas_half *>(&alpha_half), in DoBlasGemm()
1911 const Eigen::half alpha_half(alpha); in DoBlasGemmBatched() local
1916 alpha_half, a, lda, b, ldb, beta_half, c, ldc, batch_count, in DoBlasGemmBatched()
|