Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DLinearAlgebra.cpp1761 auto batch_items_contiguous_or_transposed = [&](const Tensor& t) { in bmm_out_or_baddbmm_() local
1795 && batch_items_contiguous_or_transposed(batch1) in bmm_out_or_baddbmm_()
1796 && batch_items_contiguous_or_transposed(batch2) in bmm_out_or_baddbmm_()