Home
last modified time | relevance | path

Searched defs:m_cpu (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/core/boxing/impl/
H A Dkernel_stackbased_test.cpp107 auto m_cpu = MAKE_TORCH_LIBRARY_IMPL(_test, CPU); in TEST() local
184 auto m_cpu = MAKE_TORCH_LIBRARY_IMPL(_, CPU); in TEST() local
H A Dkernel_function_test.cpp98 auto m_cpu = MAKE_TORCH_LIBRARY_IMPL(_test, CPU); in TEST() local
H A Dkernel_lambda_test.cpp77 auto m_cpu = MAKE_TORCH_LIBRARY_IMPL(_test, CPU); in TEST() local
/aosp_15_r20/external/pytorch/aten/src/ATen/core/op_registration/
H A Dop_registration_test.cpp2166 auto m_cpu = MAKE_TORCH_LIBRARY_IMPL(_test, CPU); in TEST() local
2187 auto m_cpu = MAKE_TORCH_LIBRARY_IMPL(_test, CPU); in TEST() local
2216 auto m_cpu = MAKE_TORCH_LIBRARY_IMPL(_test, CPU); in TEST() local
/aosp_15_r20/external/pytorch/test/
H A Dtest_mps.py10537 def helper(m_cpu, memory_format): argument