Searched refs:backend_fallback_kernel (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/core/op_registration/ |
| H A D | op_registration_test.cpp | 419 void backend_fallback_kernel(const c10::OperatorHandle& op, c10::Stack* stack) { in backend_fallback_kernel() function
424 …c10::DispatchKey::CPU, c10::KernelFunction::makeFromBoxedFunction<&backend_fallback_kernel>(), ""); in TEST()
434 …10::DispatchKey::CUDA, c10::KernelFunction::makeFromBoxedFunction<&backend_fallback_kernel>(), ""); in TEST()
447 …c10::DispatchKey::CPU, c10::KernelFunction::makeFromBoxedFunction<&backend_fallback_kernel>(), ""); in TEST()
462 …c10::DispatchKey::CPU, c10::KernelFunction::makeFromBoxedFunction<&backend_fallback_kernel>(), ""); in TEST()
478 …c10::DispatchKey::CPU, c10::KernelFunction::makeFromBoxedFunction<&backend_fallback_kernel>(), ""); in TEST()
1352 m1.fallback(CppFunction::makeFromBoxedFunction<&backend_fallback_kernel>()); in TEST()
1932 m.fallback(CppFunction::makeFromBoxedFunction<&backend_fallback_kernel>()); in TEST()
1988 m.def("fn6(Tensor x) -> Tensor", CppFunction::makeFromBoxedFunction<&backend_fallback_kernel>()); in TEST()
|