| /external/pytorch/torch/onnx/_internal/exporter/ |
| D | _decomp.py | 18 ) -> list[torch._ops.OperatorBase]:
28 registered_ops: list[torch._ops.OperatorBase] = []
71 onnx_registered_ops: set[torch._ops.OperatorBase], argument
72 ) -> dict[torch._ops.OperatorBase, Callable]:
86 decomposition_table: dict[torch._ops.OperatorBase, Callable] = {}
|
| /external/pytorch/torch/onnx/_internal/fx/ |
| D | decomposition_table.py | 16 ) -> set[torch._ops.OperatorBase | Callable]:
26 table: set[torch._ops.OperatorBase | Callable] = set()
75 ) -> dict[torch._ops.OperatorBase, Callable]:
89 decomposition_table: dict[torch._ops.OperatorBase, Callable] = {}
|
| /external/pytorch/torch/_decomp/ |
| D | __init__.py | 31 str, Dict[torch._ops.OperatorBase, Callable]
46 overloads: List[Union[torch._ops.OperatorBase]] = []
198 aten_ops: Sequence[Union[torch._ops.OperatorBase, OpOverloadPacket]], argument
200 ) -> Dict[torch._ops.OperatorBase, Callable]:
219 decompositions: Dict[torch._ops.OperatorBase, Callable] = {}
224 elif isinstance(op, (torch._ops.OperatorBase)) and op in registry:
230 decompositions: Dict[torch._ops.OperatorBase, Callable], argument
258 def core_aten_decompositions() -> Dict[torch._ops.OperatorBase, Callable]:
|
| D | decompositions_for_jvp.py | 13 decomposition_table_for_jvp: Dict[torch._ops.OperatorBase, Callable] = {}
|
| /external/pytorch/torch/_higher_order_ops/ |
| D | utils.py | 10 from torch._ops import OperatorBase
21 operator: OperatorBase, delayed_error: bool, *args: Any, **kwargs: Any argument
62 def autograd_not_implemented(op: OperatorBase, deferred_error: bool) -> Callable: argument
|
| D | auto_functionalize.py | 11 from torch._ops import HigherOrderOperator, OperatorBase, OpOverload
254 def can_auto_functionalize(op: OperatorBase) -> bool: argument
|
| /external/executorch/backends/transforms/ |
| D | fuse_dequant_linear.py | 56 self, node: torch.fx.Node, target: torch._ops.OperatorBase argument
|
| D | view_copy_to_squeeze_unsqueeze.py | 32 self, node: torch.fx.Node, target: torch._ops.OperatorBase argument
|
| /external/pytorch/torch/ |
| D | _ops.py | 40 class OperatorBase: class
177 def resolve_key(op: OperatorBase, k: DispatchKey): # type: ignore[valid-type] argument
242 class HigherOrderOperator(OperatorBase, abc.ABC):
649 class OpOverload(OperatorBase):
|
| /external/pytorch/torch/export/ |
| D | exported_program.py | 294 decomp_table: Dict[torch._ops.OperatorBase, Callable], argument
613 decomp_table: Dict[torch._ops.OperatorBase, Callable], argument
987 decomp_table: Optional[Dict[torch._ops.OperatorBase, Callable]] = None, argument
|
| /external/executorch/backends/qualcomm/utils/ |
| D | utils.py | 282 def get_decomp_table() -> Dict[torch._ops.OperatorBase, Callable]:
|
| /external/pytorch/torch/_inductor/ |
| D | decomposition.py | 115 ops: List[Union[torch._ops.OperatorBase, torch._ops.OpOverloadPacket]] argument
|