Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/_decomp/
H A D__init__.py198 aten_ops: Sequence[Union[torch._ops.OperatorBase, OpOverloadPacket]],
230 decompositions: Dict[torch._ops.OperatorBase, Callable],
/aosp_15_r20/external/pytorch/torch/export/
H A Dexported_program.py294 decomp_table: Dict[torch._ops.OperatorBase, Callable],
613 decomp_table: Dict[torch._ops.OperatorBase, Callable],
987 decomp_table: Optional[Dict[torch._ops.OperatorBase, Callable]] = None,
/aosp_15_r20/external/pytorch/torch/_higher_order_ops/
H A Dutils.py21 operator: OperatorBase, delayed_error: bool, *args: Any, **kwargs: Any
62 def autograd_not_implemented(op: OperatorBase, deferred_error: bool) -> Callable:
H A Dauto_functionalize.py254 def can_auto_functionalize(op: OperatorBase) -> bool:
/aosp_15_r20/external/pytorch/torch/onnx/_internal/exporter/
H A D_decomp.py71 onnx_registered_ops: set[torch._ops.OperatorBase],
/aosp_15_r20/external/executorch/backends/transforms/
H A Dfuse_dequant_linear.py56 self, node: torch.fx.Node, target: torch._ops.OperatorBase
H A Dview_copy_to_squeeze_unsqueeze.py32 self, node: torch.fx.Node, target: torch._ops.OperatorBase
/aosp_15_r20/external/pytorch/torch/
H A D_ops.py40 class OperatorBase: class
177 def resolve_key(op: OperatorBase, k: DispatchKey): # type: ignore[valid-type]
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Ddecomposition.py115 ops: List[Union[torch._ops.OperatorBase, torch._ops.OpOverloadPacket]]