Searched refs:torch_compile_op_lib_impl (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_torchinductor.py | 236 def register_ops_with_aoti_compile(ns, op_set, dispatch_key, torch_compile_op_lib_impl): argument
246 torch_compile_op_lib_impl._impl_with_aoti_compile( # noqa: F821
784 with _scoped_library("aten", "IMPL") as torch_compile_op_lib_impl:
800 ns, [op_name], dispatch_key, torch_compile_op_lib_impl
851 with _scoped_library("aten", "IMPL") as torch_compile_op_lib_impl:
856 ns, [op_name], dispatch_key, torch_compile_op_lib_impl
958 with _scoped_library("aten", "IMPL") as torch_compile_op_lib_impl:
968 namespace_name, [op_name], dispatch_key, torch_compile_op_lib_impl
1001 with _scoped_library("aten", "IMPL") as torch_compile_op_lib_impl:
1003 namespace_name, unary_op_set, dispatch_key, torch_compile_op_lib_impl
[all …]
|