You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
31 lines
756 B
31 lines
756 B
from torch._functorch import config
|
|
from torch._functorch.aot_autograd import (
|
|
aot_function,
|
|
aot_module,
|
|
aot_module_simplified,
|
|
compiled_function,
|
|
compiled_module,
|
|
get_aot_compilation_context,
|
|
get_aot_graph_name,
|
|
get_graph_being_compiled,
|
|
make_boxed_compiler,
|
|
make_boxed_func,
|
|
)
|
|
from torch._functorch.compilers import (
|
|
debug_compile,
|
|
default_decompositions,
|
|
draw_graph_compile,
|
|
memory_efficient_fusion,
|
|
nnc_jit,
|
|
nop,
|
|
print_compile,
|
|
ts_compile,
|
|
)
|
|
from torch._functorch.fx_minifier import minifier
|
|
from torch._functorch.partitioners import (
|
|
default_partition,
|
|
draw_graph,
|
|
min_cut_rematerialization_partition,
|
|
)
|
|
from torch._functorch.python_key import pythonkey_decompose
|