Home
last modified time | relevance | path

Searched defs:lowered_module (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_higher_order_ops/
H A Dexecutorch_call_delegate.py32 def __call__(self, lowered_module, *args): argument
46 def trace_call_delegate(proxy_mode, func_overload, lowered_module, *args): argument
75 def call_delegate_cpu(lowered_module, *args): argument
92 def call_delegate_autograd(lowered_module, *args): argument
124 def call_delegate_proxy_torch_dispatch_mode(mode, lowered_module, *args): argument
131 def call_delegate_fake_tensor_mode(mode, lowered_module, *args): argument
138 def call_delegate_functionalize(ctx, lowered_module, *args): argument
/aosp_15_r20/external/executorch/exir/
H A Ddelegate.py45 def trace_call_delegate(proxy_mode, func_overload, lowered_module, *args): argument
79 def call_delegate_cpu(lowered_module, *args): argument
95 def call_delegate_autograd(lowered_module, *args): argument
126 def call_delegate_proxy_torch_dispatch_mode(mode, lowered_module, *args): argument
132 def call_delegate_fake_tensor_mode(mode, lowered_module, *args): argument
138 def call_delegate_functionalize(ctx, lowered_module, *args): argument
/aosp_15_r20/external/executorch/exir/backend/test/
H A Dtest_lowered_backend_module.py49 self, lowered_module, example_inputs, edge_compile_config argument
200 def __init__(self, lowered_module): argument
/aosp_15_r20/external/executorch/examples/apple/coreml/scripts/
H A Dexport.py107 def export_lowered_module_to_executorch_program(lowered_module, example_inputs): argument
/aosp_15_r20/external/executorch/examples/apple/mps/scripts/
H A Dmps_example.py189 lowered_module = to_backend( variable
/aosp_15_r20/external/executorch/exir/passes/
H A Dspec_prop_pass.py131 def call_delegate(self, lowered_module, args, kwargs, meta): argument
/aosp_15_r20/external/executorch/docs/source/tutorials_source/
H A Dexport-to-executorch-tutorial.py372 lowered_module: LoweredBackendModule = to_backend( variable
/aosp_15_r20/external/executorch/backends/arm/test/tester/
H A Darm_tester.py64 def get_output_format(lowered_module) -> str | None: argument