Home
last modified time | relevance | path

Searched refs:py_functionalize_impl (Results 1 – 18 of 18) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_higher_order_ops/
H A Drun_const_graph.py36 @run_const_graph.py_functionalize_impl
H A Dstrict_mode.py87 @strict_mode_op.py_functionalize_impl
H A Dauto_functionalize.py613 @auto_functionalized.py_functionalize_impl
708 @auto_functionalized_v2.py_functionalize_impl
H A Dtorchbind.py136 @call_torchbind.py_functionalize_impl
H A Dhints_wrap.py92 @hints_wrapper.py_functionalize_impl
H A Dout_dtype.py160 @out_dtype.py_functionalize_impl
H A Dexecutorch_call_delegate.py136 @executorch_call_delegate.py_functionalize_impl
H A Dwhile_loop.py236 @while_loop_op.py_functionalize_impl
H A Dmap.py243 @map_impl.py_functionalize_impl
H A Dtriton_kernel_wrap.py627 @triton_kernel_wrapper_mutation.py_functionalize_impl
719 @triton_kernel_wrapper_functional.py_functionalize_impl
H A Dassociative_scan.py357 @associative_scan_op.py_functionalize_impl
H A Dflex_attention.py386 @flex_attention.py_functionalize_impl
952 @flex_attention_backward.py_functionalize_impl
H A Dcond.py448 @cond_op.py_functionalize_impl
/aosp_15_r20/external/pytorch/torch/_dynamo/
H A D_trace_wrapped_higher_order_op.py123 @_trace_wrapped_op.py_functionalize_impl
/aosp_15_r20/external/executorch/exir/
H A Ddelegate.py136 @executorch_call_delegate.py_functionalize_impl
/aosp_15_r20/external/pytorch/torch/_export/
H A Dwrappers.py41 @_export_tracepoint.py_functionalize_impl
/aosp_15_r20/external/pytorch/torch/_prims/
H A Drng_prims.py299 @run_with_rng_state.py_functionalize_impl
/aosp_15_r20/external/pytorch/torch/
H A D_ops.py144 def py_functionalize_impl(self, fn): member in OperatorBase