Searched refs:py_functionalize_impl (Results 1 – 18 of 18) sorted by relevance
36 @run_const_graph.py_functionalize_impl
87 @strict_mode_op.py_functionalize_impl
613 @auto_functionalized.py_functionalize_impl708 @auto_functionalized_v2.py_functionalize_impl
136 @call_torchbind.py_functionalize_impl
92 @hints_wrapper.py_functionalize_impl
160 @out_dtype.py_functionalize_impl
136 @executorch_call_delegate.py_functionalize_impl
236 @while_loop_op.py_functionalize_impl
243 @map_impl.py_functionalize_impl
627 @triton_kernel_wrapper_mutation.py_functionalize_impl719 @triton_kernel_wrapper_functional.py_functionalize_impl
357 @associative_scan_op.py_functionalize_impl
386 @flex_attention.py_functionalize_impl952 @flex_attention_backward.py_functionalize_impl
448 @cond_op.py_functionalize_impl
123 @_trace_wrapped_op.py_functionalize_impl
41 @_export_tracepoint.py_functionalize_impl
299 @run_with_rng_state.py_functionalize_impl
144 def py_functionalize_impl(self, fn): member in OperatorBase