Searched defs:batch_stride_a (Results 1 – 4 of 4) sorted by relevance
149 int64_t batch_stride_a = in GetPlanAndAlgorithms() local
564 const scalar_t *a, int64_t lda, int64_t batch_stride_a, in gemm_batched_with_stride_generic()581 const scalar_t *a, int64_t lda, int64_t batch_stride_a, in gemm_batched_with_stride()
482 … int batch_stride_a = a->info()->strides_in_bytes()[a_batch_idx] / a->info()->element_size(); in run() local
775 uint64_t batch_stride_a = 0; in DoBlasGemmBatchedInternal() local