Home
last modified time | relevance | path

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

/external/pytorch/torch/_higher_order_ops/
Dauto_functionalize.py359 with ctx.redispatch_to_next():
499 with ctx.redispatch_to_next():
616 with ctx.redispatch_to_next():
711 with ctx.redispatch_to_next():
Drun_const_graph.py40 with ctx.redispatch_to_next():
Dstrict_mode.py90 with ctx.redispatch_to_next():
Dhints_wrap.py97 with ctx.redispatch_to_next():
Dout_dtype.py164 with ctx.redispatch_to_next():
Dexecutorch_call_delegate.py140 with ctx.redispatch_to_next():
Dwhile_loop.py241 with ctx.redispatch_to_next() as m:
Deffects.py268 with ctx.redispatch_to_next():
Dmap.py249 with ctx.redispatch_to_next():
Dassociative_scan.py360 with ctx.redispatch_to_next() as m:
Dcond.py452 with ctx.redispatch_to_next() as m:
Dtriton_kernel_wrap.py641 with ctx.redispatch_to_next():
724 with ctx.redispatch_to_next():
Dflex_attention.py429 with ctx.redispatch_to_next() as m:
1003 with ctx.redispatch_to_next() as m:
/external/pytorch/torch/_dynamo/
D_trace_wrapped_higher_order_op.py126 with ctx.redispatch_to_next():
/external/pytorch/torch/_subclasses/
Dfunctional_tensor.py652 def redispatch_to_next(self) -> ContextManager: member in BaseFunctionalizeAPI
696 def redispatch_to_next(self) -> ContextManager: member in PythonFunctionalizeAPI
740 def redispatch_to_next(self) -> ContextManager: member in CppFunctionalizeAPI
788 def redispatch_to_next(self) -> ContextManager: member in FunctorchFunctionalizeAPI
/external/executorch/exir/
Ddelegate.py140 with ctx.redispatch_to_next():
/external/pytorch/torch/_export/
Dwrappers.py46 with ctx.redispatch_to_next():
/external/pytorch/torch/_prims/
Drng_prims.py305 with ctx.redispatch_to_next():