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