/aosp_15_r20/external/pytorch/torch/_higher_order_ops/ |
H A D | executorch_call_delegate.py | 32 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 D | delegate.py | 45 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 D | test_lowered_backend_module.py | 49 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 D | export.py | 107 def export_lowered_module_to_executorch_program(lowered_module, example_inputs): argument
|
/aosp_15_r20/external/executorch/examples/apple/mps/scripts/ |
H A D | mps_example.py | 189 lowered_module = to_backend( variable
|
/aosp_15_r20/external/executorch/exir/passes/ |
H A D | spec_prop_pass.py | 131 def call_delegate(self, lowered_module, args, kwargs, meta): argument
|
/aosp_15_r20/external/executorch/docs/source/tutorials_source/ |
H A D | export-to-executorch-tutorial.py | 372 lowered_module: LoweredBackendModule = to_backend( variable
|
/aosp_15_r20/external/executorch/backends/arm/test/tester/ |
H A D | arm_tester.py | 64 def get_output_format(lowered_module) -> str | None: argument
|