Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DLinearAlgebra.cpp1821 const bool enable_multithreaded_bmm = c10::InferenceMode::is_enabled() && in bmm_out_or_baddbmm_() local
1824 const bool enable_multithreaded_bmm{false}; in bmm_out_or_baddbmm_()
1827 if (enable_multithreaded_bmm) { in bmm_out_or_baddbmm_()
1846 if (enable_multithreaded_bmm) { in bmm_out_or_baddbmm_()