Searched defs:batch2_sizes (Results 1 – 1 of 1) sorted by relevance
289 const auto batch2_sizes = batch2.sizes(); in common_checks_baddbmm_bmm() local1741 const auto batch2_sizes = batch2.sizes(); in bmm_out_or_baddbmm_() local