Searched defs:OpOverloadPacket (Results 1 – 11 of 11) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_decomp/ |
| H A D | __init__.py | 198 aten_ops: Sequence[Union[torch._ops.OperatorBase, OpOverloadPacket]], argument
231 aten_ops: Sequence[Union[OpOverload, OpOverloadPacket]], argument
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/optests/ |
| H A D | generate_tests.py | 622 op: Union[torch._ops.OpOverload, torch._ops.OpOverloadPacket, CustomOpDef],
724 op: torch._ops.OpOverloadPacket,
|
| /aosp_15_r20/external/pytorch/torch/onnx/_internal/fx/passes/ |
| H A D | type_promotion.py | 1238 def get_rule(self, py_op: torch._ops.OpOverloadPacket) -> TypePromotionRule | None:
1296 op: torch._ops.OpOverloadPacket, args: tuple, kwargs: dict
|
| /aosp_15_r20/external/executorch/exir/dialects/backend/ |
| H A D | _ops.py | 86 parent_overload_packet: torch._ops.OpOverloadPacket,
|
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | _ops.py | 1008 class OpOverloadPacket: class
1125 def _call_overload_packet_from_python(op: OpOverloadPacket, args, kwargs):
|
| H A D | library.py | 1214 op: Union[torch._ops.OpOverload, torch._ops.OpOverloadPacket, CustomOpDef],
|
| /aosp_15_r20/external/executorch/exir/backend/ |
| H A D | utils.py | 104 replacement_op: torch._ops.OpOverloadPacket,
|
| /aosp_15_r20/external/pytorch/torch/ao/quantization/quantizer/ |
| H A D | xnnpack_quantizer.py | 311 operator_type: torch._ops.OpOverloadPacket,
|
| H A D | x86_inductor_quantizer.py | 603 operator_type: torch._ops.OpOverloadPacket,
|
| /aosp_15_r20/external/executorch/exir/dialects/edge/ |
| H A D | _ops.py | 354 parent_overload_packet: torch._ops.OpOverloadPacket,
|
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | decomposition.py | 115 ops: List[Union[torch._ops.OperatorBase, torch._ops.OpOverloadPacket]]
|