Searched refs:b_desc (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_blas.cc | 3309 cublasLtMatrixLayout_t b_desc() const { return b_desc_.get(); } in DoBlasInternalImpl() function in stream_executor::gpu::__anone9de0a230311::CUDABlasLtMatmulPlan
3481 const auto &b_desc = in DoBlasInternalImpl() local
3482 for_remainder_batch ? cuda_plan.b_remainder_desc() : cuda_plan.b_desc(); in DoBlasInternalImpl()
3488 blasLt_, cuda_plan.op_desc(), a_desc, b_desc, c_desc, d_desc, in DoBlasInternalImpl()
3636 b_ptr, cuda_plan.b_desc(), beta_ptr, c_ptr, cuda_plan.c_desc(), d_ptr, in DoBlasInternalImpl()
|