Searched refs:capped_batch_count (Results 1 – 1 of 1) sorted by relevance
3274 p.stride_a, capped_batch_count())); in DoBlasInternalImpl()3277 p.stride_b, capped_batch_count())); in DoBlasInternalImpl()3280 capped_batch_count())); in DoBlasInternalImpl()3283 capped_batch_count())); in DoBlasInternalImpl()3329 int capped_batch_count() const { in DoBlasInternalImpl() function in stream_executor::gpu::__anone9de0a230311::CUDABlasLtMatmulPlan3630 int capped_batch_count = cuda_plan.capped_batch_count(); in DoBlasInternalImpl() local3632 batch + capped_batch_count <= cuda_plan.params().batch_count; in DoBlasInternalImpl()3633 batch += capped_batch_count) { in DoBlasInternalImpl()3645 a_ptr += capped_batch_count * cuda_plan.params().stride_a * ab_type_size; in DoBlasInternalImpl()3646 b_ptr += capped_batch_count * cuda_plan.params().stride_b * ab_type_size; in DoBlasInternalImpl()[all …]