Home
last modified time | relevance | path

Searched defs:OpOverload (Results 1 – 25 of 65) sorted by relevance

123

/aosp_15_r20/external/pytorch/torch/_higher_order_ops/
H A Dauto_functionalize.py210 _mutable_op: OpOverload,
239 _mutable_op: OpOverload,
302 def get_mutable_args(op: OpOverload) -> Tuple[List[str], List[torch.Type]]:
322 op: OpOverload,
411 op: OpOverload,
551 _mutable_op: OpOverload,
585 _mutable_op: OpOverload,
596 _mutable_op: OpOverload,
624 _mutable_op: OpOverload,
680 _mutable_op: OpOverload,
[all …]
H A Dout_dtype.py102 def out_dtype_dense(op: torch._ops.OpOverload, output_dtype: torch.dtype, *args):
142 op: torch._ops.OpOverload,
152 op: torch._ops.OpOverload,
H A Deffects.py137 op: torch._ops.OpOverload,
152 op: torch._ops.OpOverload,
165 op: torch._ops.OpOverload,
/aosp_15_r20/external/pytorch/torch/_subclasses/
H A Dfake_tensor.py214 def get_schema_info(func: OpOverload) -> torch._C._SchemaInfo:
224 def torch_decomp_decompositions(func: OpOverload) -> bool:
514 def should_allow_numbers_as_tensors(func: OpOverload) -> bool:
750 func: OpOverload,
823 func: OpOverload, flat_args: Sequence[object]
1228 func: OpOverload,
1311 func: OpOverload,
1355 func: OpOverload,
1388 func: OpOverload,
1478 func: OpOverload,
[all …]
H A Dfake_impls.py115 def _is_tensor_constructor(func: OpOverload):
126 def register_op_impl(run_impl_check: Union[Callable[[OpOverload], bool], OpOverload]):
751 def register_fast_op_impl(func: OpOverload):
/aosp_15_r20/external/pytorch/torch/distributed/tensor/
H A D_dispatch.py45 op_call: torch._ops.OpOverload,
61 op_call: torch._ops.OpOverload,
71 op_call: torch._ops.OpOverload,
154 op_call: torch._ops.OpOverload,
330 op_call: torch._ops.OpOverload,
445 op_call: torch._ops.OpOverload,
478 op_call: torch._ops.OpOverload,
H A D_tp_conv.py107 op_call: torch._ops.OpOverload,
160 op_call: torch._ops.OpOverload,
231 op_call: torch._ops.OpOverload,
254 op_call: torch._ops.OpOverload,
H A D_sharding_prop.py83 op_overload: OpOverload,
96 op_overload: OpOverload,
155 op: OpOverload,
H A D_op_schema.py46 def _is_inplace_op(op: OpOverload):
53 def _is_out_variant_op(op: OpOverload):
/aosp_15_r20/external/executorch/exir/operator/
H A Dconvert.py85 def _get_overload_schema(op_overload: OpOverload) -> Optional[FunctionSchema]:
93 def get_out_args_from_opoverload(op_overload: OpOverload) -> Tuple[str]:
137 def set_mapping_for_op(op: OpOverload) -> None:
231 def to_out_variant(op_overload: OpOverload) -> Tuple[OpOverload, Tuple[str]]:
275 def to_scratch_op(op_overload: OpOverload) -> Optional[OpOverload]:
H A Dmanip.py32 out_variant: torch._ops.OpOverload,
78 def attach_calculate_upper_bound_shape_fn(func_op: torch._ops.OpOverload):
/aosp_15_r20/external/pytorch/torch/_custom_op/
H A Dfunctional.py24 mutable_op: OpOverload,
146 def validate(mutable_op: OpOverload):
180 def functional_schema(new_op_name, op: OpOverload):
186 def mutable_args(op: OpOverload):
/aosp_15_r20/external/pytorch/torch/distributed/tensor/parallel/
H A Dloss.py109 op_call: torch._ops.OpOverload,
155 op_call: torch._ops.OpOverload,
186 op_call: torch._ops.OpOverload,
271 op_call: torch._ops.OpOverload,
416 op_call: torch._ops.OpOverload,
/aosp_15_r20/external/pytorch/torch/fx/experimental/
H A Dproxy_tensor.py144 decomposition_table: Optional[Mapping[OpOverload, Callable]]
762 func: OpOverload,
1202 def set_original_aten_op(func: OpOverload) -> Generator[None, None, None]:
1222 func: OpOverload,
1242 func: OpOverload,
1297 func: OpOverload,
1336 self, func: OpOverload, args: Tuple[object, ...], out: PySymType
1354 func: OpOverload,
1417 decomposition_table: Optional[Mapping[OpOverload, Callable]] = None,
1783 decomposition_table: Optional[Mapping[OpOverload, Callable]],
[all …]
/aosp_15_r20/external/pytorch/torch/testing/_internal/optests/
H A Dgenerate_tests.py47 op: torch._ops.OpOverload,
63 op: torch._ops.OpOverload,
82 op: torch._ops.OpOverload,
96 op: torch._ops.OpOverload,
622 op: Union[torch._ops.OpOverload, torch._ops.OpOverloadPacket, CustomOpDef],
676 op: torch._ops.OpOverload,
/aosp_15_r20/external/pytorch/torch/_library/
H A Dutils.py70 def is_builtin(op: OpOverload) -> bool:
124 def mutates_and_returns_first_arg(op: OpOverload):
242 def can_generate_trivial_fake_impl(op: OpOverload) -> bool:
/aosp_15_r20/external/executorch/backends/qualcomm/quantizer/
H A Dquantizer.py125 def _get_quant_config(self, op: str | OpOverload) -> Optional[QuantizationConfig]:
142 def _update_per_channel_weight_quant_ops(self, ops: Set[OpOverload], enable: bool):
156 def add_discard_ops(self, ops: Sequence[OpOverload]) -> None:
/aosp_15_r20/external/pytorch/torch/distributed/tensor/experimental/
H A D_register_sharding.py24 def register_sharding(op: Union[OpOverload, List[OpOverload]]):
/aosp_15_r20/external/pytorch/torch/export/
H A Dexported_program.py295 _preserve_ops: Tuple[torch._ops.OpOverload],
614 _preserve_ops: Tuple[torch._ops.OpOverload],
988 _preserve_ops: Tuple[torch._ops.OpOverload, ...] = (),
/aosp_15_r20/external/executorch/exir/verification/
H A Dverifier.py80 exception_list: Optional[List[torch._ops.OpOverload]] = None,
188 exception_list: Optional[List[torch._ops.OpOverload]] = None,
/aosp_15_r20/external/executorch/exir/program/
H A D_program.py706 ops_set_to_not_decompose: Optional[List[torch._ops.OpOverload]] = None,
1074 preserve_ops: Tuple[torch._ops.OpOverload, ...] = (),
1165 ops_set_to_not_decompose: Optional[List[torch._ops.OpOverload]] = None,
/aosp_15_r20/external/executorch/exir/serde/
H A Dexport_serialize.py383 def _is_single_tensor_return(target: torch._ops.OpOverload) -> bool:
388 def _is_single_tensor_list_return(target: torch._ops.OpOverload) -> bool:
617 self, target: torch._ops.OpOverload, args, kwargs=None
1897 def deserialize_inputs(self, target: torch._ops.OpOverload, serialized_node: Node):
/aosp_15_r20/external/pytorch/torch/distributed/_symmetric_memory/
H A D__init__.py288 mm_out_op: torch._ops.OpOverload,
546 mm_out_op: torch._ops.OpOverload,
/aosp_15_r20/external/pytorch/torch/jit/
H A D_decomposition_utils.py6 def _register_decomposition(op: OpOverload, graph: torch._C.Graph):
/aosp_15_r20/external/executorch/exir/passes/
H A Dreplace_edge_with_backend_pass.py40 target: torch._ops.OpOverload,

123