Home
last modified time | relevance | path

Searched defs:batch_stride_b (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dmatmul_util.cc151 int64_t batch_stride_b = in GetPlanAndAlgorithms() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DCPUBlas.cpp565 const scalar_t *b, int64_t ldb, int64_t batch_stride_b, in gemm_batched_with_stride_generic()
582 const scalar_t *b, int64_t ldb, int64_t batch_stride_b, in gemm_batched_with_stride()
/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/
H A Drocm_blas.cc776 uint64_t batch_stride_b = 0; in DoBlasGemmBatchedInternal() local