Home
last modified time | relevance | path

Searched defs:ProxyValue (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/executorch/exir/passes/
H A Ddynamic_shape_prop_pass.py92 inputs: Union[List[ProxyValue], Tuple[ProxyValue, ...]],
109 def inject_xs_dsinfo_to_graph(self, subgm: GraphModule, xs: ProxyValue):
154 pred: ProxyValue,
174 xs: ProxyValue,
175 args: Tuple[ProxyValue, ...],
H A Dspec_prop_pass.py53 def on_attr(self, attr: ProxyValue) -> None:
115 mapped_args: List[ProxyValue],
116 operands: List[ProxyValue],
/aosp_15_r20/external/executorch/exir/
H A Dpass_base.py118 class ProxyValue: class
454 def on_attr(self, attr: ProxyValue) -> None:
482 pred: ProxyValue,
503 mapped_args: List[ProxyValue],
504 operands: List[ProxyValue],
519 self, value: ProxyValue, key: int, meta: NodeMetadata
646 args: Tuple[ProxyValue, ...],
/aosp_15_r20/external/pytorch/torch/_export/
H A Dpass_base.py317 def on_attr(self, attr: ProxyValue) -> None:
345 pred: ProxyValue,
366 mapped_args: List[ProxyValue],
367 operands: List[ProxyValue],
382 self, value: ProxyValue, key: int, meta: NodeMetadata
/aosp_15_r20/external/executorch/backends/cadence/aot/
H A Dreplace_ops.py1127 self: ExportPass, proxy: ProxyValue, meta: NodeMetadata, dim0: int, dim1: int
1143 def change_nchw_to_nhwc(self, proxy: ProxyValue, meta: NodeMetadata) -> ProxyValue:
1153 def change_nhwc_to_nchw(self, proxy: ProxyValue, meta: NodeMetadata) -> ProxyValue:
H A Dremove_ops.py172 args: tuple[ProxyValue, tuple[int, ...], Argument],
376 args: tuple[ProxyValue],
H A Dgraph_builder.py60 def output(self, results: list[ProxyValue]) -> ProxyValue:
/aosp_15_r20/external/pytorch/torch/_export/pass_infra/
H A Dproxy_value.py8 class ProxyValue: class
/aosp_15_r20/external/executorch/exir/tests/
H A Dtest_memory_format_ops_pass.py112 def grab_tensor(arg: Union[ProxyValue, torch.Tensor]):