Searched refs:custom_cpu_fallback (Results 1 – 1 of 1) sorted by relevance
524 void custom_cpu_fallback(const c10::OperatorHandle& op, torch::jit::Stack* stack) { in custom_cpu_fallback() function529 m.impl("sub.Tensor", torch::CppFunction::makeFromBoxedFunction<&custom_cpu_fallback>()); in TORCH_LIBRARY_IMPL()530 m.impl("_foreach_add.List", torch::CppFunction::makeFromBoxedFunction<&custom_cpu_fallback>()); in TORCH_LIBRARY_IMPL()531 m.impl("_fused_adamw_", torch::CppFunction::makeFromBoxedFunction<&custom_cpu_fallback>()); in TORCH_LIBRARY_IMPL()532 m.impl("index.Tensor", torch::CppFunction::makeFromBoxedFunction<&custom_cpu_fallback>()); in TORCH_LIBRARY_IMPL()533 m.impl("triu_indices", torch::CppFunction::makeFromBoxedFunction<&custom_cpu_fallback>()); in TORCH_LIBRARY_IMPL()