Searched defs:aoti_custom_ops (Results 1 – 2 of 2) sorted by relevance
304 TORCH_LIBRARY(aoti_custom_ops, m) { in TORCH_LIBRARY() argument347 TORCH_LIBRARY_IMPL(aoti_custom_ops, CompositeExplicitAutograd, m) { in TORCH_LIBRARY_IMPL() argument357 TORCH_LIBRARY_IMPL(aoti_custom_ops, Meta, m) { in TORCH_LIBRARY_IMPL() argument
3 add_library(aoti_custom_ops SHARED custom_ops.cpp) target