Searched refs:beta_ptr (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_blas.cc | 3564 const void *beta_ptr = in DoBlasInternalImpl() local
3636 b_ptr, cuda_plan.b_desc(), beta_ptr, c_ptr, cuda_plan.c_desc(), d_ptr, in DoBlasInternalImpl()
3672 beta_ptr, c_ptr, cuda_plan.c_remainder_desc(), d_ptr, in DoBlasInternalImpl()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_blas.cc | 2115 MAPPED_T *beta_ptr = reinterpret_cast<MAPPED_T *>(&beta); in DoBlasGemmBatchedInternal() local
2122 GpuComplex(beta_ptr), GpuMemoryMutable(&c), ldc, in DoBlasGemmBatchedInternal()
|