Home
last modified time | relevance | path

Searched refs:for_remainder_batch (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_blas.h158 bool for_remainder_batch = false);
Dcuda_blas.cc3464 bool for_remainder_batch) { in DoBlasInternalImpl() argument
3480 for_remainder_batch ? cuda_plan.a_remainder_desc() : cuda_plan.a_desc(); in DoBlasInternalImpl()
3482 for_remainder_batch ? cuda_plan.b_remainder_desc() : cuda_plan.b_desc(); in DoBlasInternalImpl()
3484 for_remainder_batch ? cuda_plan.c_remainder_desc() : cuda_plan.c_desc(); in DoBlasInternalImpl()
3486 for_remainder_batch ? cuda_plan.d_remainder_desc() : cuda_plan.d_desc(); in DoBlasInternalImpl()