Home
last modified time | relevance | path

Searched defs:Callable (Results 1 – 25 of 901) sorted by relevance

12345678910>>...37

/aosp_15_r20/external/pytorch/torch/fx/passes/
H A Dpass_manager.py20 def inplace_wrapper(fn: Callable) -> Callable:
39 def log_hook(fn: Callable, level=logging.INFO) -> Callable:
80 def loop_pass(base_pass: Callable, n_iter: Optional[int] = None, predicate: Optional[Callable] = No…
120 constraint: Callable[[Callable, Callable], bool], passes: List[Callable] argument
133 def this_before_that_pass_constraint(this: Callable, that: Callable):
139 def depends_on(a: Callable, b: Callable):
145 def these_before_those_pass_constraint(these: Callable, those: Callable):
170 def depends_on(a: Callable, b: Callable):
210 def add_pass(self, _pass: Callable):
/aosp_15_r20/external/pytorch/torch/fx/passes/infra/
H A Dpass_manager.py19 def pass_result_wrapper(fn: Callable) -> Callable:
50 constraint: Callable[[Callable, Callable], bool], passes: List[Callable] argument
63 passes: List[Callable], constraints: List[Callable]
117 def this_before_that_pass_constraint(this: Callable, that: Callable) -> Callable:
139 def depends_on(a: Callable, b: Callable):
188 def add_pass(self, _pass: Callable):
195 def add_constraint(self, constraint: Callable):
225 def add_checks(self, check: Callable) -> None:
/aosp_15_r20/external/pytorch/torch/ao/ns/
H A D_numeric_suite_fx.py313 Dict[str, Dict[Callable, Callable]]
338 Dict[str, Dict[Callable, Callable]]
390 Dict[str, Dict[Callable, Callable]]
448 logger_cls: Callable,
478 logger_cls: Callable,
534 logger_cls: Callable,
588 logger_cls: Callable,
643 logger_cls: Callable,
680 logger_cls: Callable,
710 logger_cls: Callable,
[all …]
/aosp_15_r20/external/pytorch/torch/_functorch/
H A Daot_autograd.py769 fn: Callable,
770 fw_compiler: Callable,
771 bw_compiler: Optional[Callable] = None,
772 partition_fn: Callable = default_partition,
776 inference_compiler: Optional[Callable] = None,
939 fw_compiler: Callable,
940 bw_compiler: Optional[Callable] = None,
941 partition_fn: Callable = default_partition,
944 inference_compiler: Optional[Callable] = None,
1317 func: Callable,
[all …]
H A Ddeprecated.py59 func: Callable,
70 def grad(func: Callable, argnums: argnums_t = 0, has_aux: bool = False) -> Callable:
76 func: Callable, argnums: argnums_t = 0, has_aux: bool = False
82 def vjp(func: Callable, *primals, has_aux: bool = False):
88 func: Callable,
100 func: Callable,
118 func: Callable,
133 def functionalize(func: Callable, *, remove: str = "mutations") -> Callable:
H A Deager_transforms.py242 def vjp(func: Callable, *primals, has_aux: bool = False):
373 func: Callable, *primals, argnums: Optional[argnums_t] = None, has_aux: bool = False
466 func: Callable,
1026 func: Callable,
1089 func: Callable,
1179 func: Callable,
1448 def grad_impl(func: Callable, argnums: argnums_t, has_aux: bool, args, kwargs):
1508 def functionalize(func: Callable, *, remove: str = "mutations") -> Callable:
1723 def linearize(func: Callable, *primals) -> Tuple[Any, Callable]:
/aosp_15_r20/external/pytorch/torch/
H A D_vmap_internals.py44 error_message_lambda: Callable[[], str],
59 func: Callable,
121 func: Callable,
158 def _validate_outputs(outputs: Any, func: Callable) -> None:
175 def _check_out_dims_is_int_or_int_tuple(out_dims: out_dims_t, func: Callable) -> None:
188 def _get_name(func: Callable):
205 def vmap(func: Callable, in_dims: in_dims_t = 0, out_dims: out_dims_t = 0) -> Callable:
214 func: Callable,
/aosp_15_r20/external/pytorch/torch/_higher_order_ops/
H A Dflex_attention.py97 score_mod: Callable,
138 fw_graph: Union[Callable, GraphModule],
176 score_mod: Callable,
217 score_mod: Callable,
269 score_mod: Callable,
296 score_mod: Callable,
364 score_mod: Callable,
392 score_mod: Callable,
461 score_mod: Callable,
685 score_mod: Callable,
[all …]
/aosp_15_r20/external/pytorch/torch/utils/data/_utils/
H A Dcollate.py121 collate_fn_map: Optional[Dict[Union[Type, Tuple[Type, ...]], Callable]] = None, argument
246 collate_fn_map: Optional[Dict[Union[Type, Tuple[Type, ...]], Callable]] = None, argument
278 collate_fn_map: Optional[Dict[Union[Type, Tuple[Type, ...]], Callable]] = None, argument
291 collate_fn_map: Optional[Dict[Union[Type, Tuple[Type, ...]], Callable]] = None, argument
299 collate_fn_map: Optional[Dict[Union[Type, Tuple[Type, ...]], Callable]] = None, argument
307 collate_fn_map: Optional[Dict[Union[Type, Tuple[Type, ...]], Callable]] = None, argument
315 collate_fn_map: Optional[Dict[Union[Type, Tuple[Type, ...]], Callable]] = None, argument
/aosp_15_r20/external/pytorch/torch/_library/
H A Dcustom_ops.py35 fn: Optional[Callable] = None,
170 def __init__(self, namespace: str, name: str, schema: str, fn: Callable) -> None:
268 self, device_types: device_types_t, fn: Optional[Callable] = None, /
387 def register_fake(self, fn: Callable, /) -> Callable:
462 self, torch_dispatch_class: Any, fn: Optional[Callable] = None, /
494 backward: Callable,
497 setup_context: Optional[Callable] = None,
673 func: Optional[Callable] = None,
/aosp_15_r20/external/executorch/exir/
H A Dtracer.py189 wrapper: Callable[..., LeafValue],
200 wrapper: Callable[..., LeafValue],
201 meta_type: Callable[..., Iterable[ValueSpec]] = tuple,
414 forward: Callable[..., Value],
482 root: Callable[..., Value],
494 root: Callable[..., Value],
573 f: Callable[..., Value],
639 f: Callable[..., Value],
693 f: Callable[..., Value],
H A Dcontrol_flow.py96 fn: Callable[..., Union[torch.Tensor, Tuple[torch.Tensor]]],
130 cond_fn: Callable[..., torch.Tensor],
131 body_fn: Callable[..., Tuple[torch.Tensor]],
188 f: Callable[..., Tuple[torch.Tensor]]
/aosp_15_r20/external/python/pyee/pyee/
Dbase.py111 def _add_event_handler(self, event: str, k: Callable, v: Callable):
126 f: Callable,
186 f: Callable = None,
192 def _wrapper(f: Callable) -> Callable:
216 def _remove_listener(self, event: str, f: Callable) -> None:
222 def remove_listener(self, event: str, f: Callable) -> None:
/aosp_15_r20/external/pytorch/torch/utils/benchmark/utils/
H A Dcompile.py41 model: Union[torch.nn.Module, Callable], argument
45 loss_fn: Optional[Callable] = None,
75 model: Union[torch.nn.Module, Callable], argument
120 model : Union[torch.nn.Module, Callable], argument
/aosp_15_r20/external/pytorch/torch/fx/
H A Dsubgraph_rewriter.py85 pattern: Union[Callable, GraphModule],
86 replacement: Union[Callable, GraphModule]
209 pattern: Union[Callable, Graph, GraphModule],
210 replacement: Union[Callable, Graph, GraphModule],
229 pattern: Union[Callable, Graph, GraphModule],
230 replacement: Union[Callable, Graph, GraphModule],
/aosp_15_r20/external/llvm-libc/test/UnitTest/
H A DFPExceptMatcher.h34 struct Callable : public FunctionCaller { in getFunctionCaller() struct
35 Func f; in getFunctionCaller()
36 explicit Callable(Func theFunc) : f(theFunc) {} in getFunctionCaller() function
37 void call() override { f(); } in getFunctionCaller()
/aosp_15_r20/external/pytorch/torch/fx/experimental/
H A Dproxy_tensor.py144 decomposition_table: Optional[Mapping[OpOverload, Callable]] argument
336 transform: Callable = lambda x: x,
1035 forward: Callable[..., Any],
1123 root: Union[Module, Callable], argument
1417 decomposition_table: Optional[Mapping[OpOverload, Callable]] = None, argument
1468 func: Callable[..., R], args: Tuple[object, ...], kwargs: Dict[str, object]
1629 self, root: Union[Module, Callable], concrete_args: Optional[Dict[str, object]] argument
1681 forward: Callable,
1783 decomposition_table: Optional[Mapping[OpOverload, Callable]], argument
1849 self, f: Callable, args: Tuple[object, ...]
[all …]
/aosp_15_r20/external/grpc-grpc/src/python/grpcio/grpc/
H A D_common.py112 wait_fn: Callable[..., bool],
114 spin_cb: Optional[Callable[[], None]],
122 wait_fn: Callable[..., bool],
123 wait_complete_fn: Callable[[], bool],
125 spin_cb: Optional[Callable[[], None]] = None,
H A D_interceptor.py36 def _continuation(self, thunk: Callable, index: int) -> Callable:
40 self, thunk: Callable, index: int, context: grpc.HandlerCallDetails
50 self, thunk: Callable, context: grpc.HandlerCallDetails
260 thunk: Callable,
403 thunk: Callable,
462 thunk: Callable,
605 thunk: Callable,
680 self, callback: Callable, try_to_connect: Optional[bool] = False
684 def unsubscribe(self, callback: Callable):
/aosp_15_r20/external/pytorch/torch/ao/quantization/pt2e/
H A Dqat_utils.py86 def _get_conv_bn_pattern(conv_fn: Callable) -> Callable:
106 def _get_qat_conv_bn_pattern(conv_fn: Callable) -> Callable:
149 def _get_qat_conv_bn_pattern_no_conv_bias(conv_fn: Callable) -> Callable:
221 conv_fn: Callable,
289 conv_fn: Callable,
655 conv_fn: Callable,
876 conv_fn: Callable,
/aosp_15_r20/external/pytorch/torch/ao/quantization/
H A Dquantization_mappings.py239 float_module_class: Callable,
240 additional_static_quant_mapping: Optional[Dict[Callable, Any]] = None,
263 float_module_class: Callable,
264 additional_dynamic_quant_mapping: Optional[Dict[Callable, Any]] = None,
344 def get_quantized_operator(float_op: Union[Callable, str]) -> Callable:
/aosp_15_r20/external/pytorch/torch/onnx/_internal/
H A Dregistration.py160 def add(self, func: Callable, opset: OpsetVersion) -> None:
176 def add_custom(self, func: Callable, opset: OpsetVersion) -> None:
215 self, name: str, opset: OpsetVersion, func: Callable, custom: bool = False
271 decorate: Optional[Sequence[Callable]] = None,
299 def wrapper(func: Callable) -> Callable:
322 decorate: Optional[Sequence[Callable]] = None,
/aosp_15_r20/external/toolchain-utils/llvm_tools/
H A Dpatch_utils.py272 patch_cmd: Optional[Callable] = None,
312 self, root_dir: Path, patch_cmd: Optional[Callable] = None
409 patch_cmd: Optional[Callable] = None,
462 patch_cmd: Optional[Callable] = None,
552 patch_cmd: Optional[Callable] = None,
603 patch_cmd: Optional[Callable] = None,
/aosp_15_r20/external/pytorch/torch/_inductor/fx_passes/
H A Dddp_fusion.py52 target: Union[str, Callable[..., Any]], argument
164 comm_filter: Optional[Callable[..., bool]] = None,
476 graph: fx.Graph, algorithm_fn: Callable[..., Any], fusion_fn: Callable[..., Any]
582 graph: fx.Graph, passes: List[Union[Callable[..., None], str]], bucket_size_mb: int
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dcommon_fsdp.py114 assert_fn: Callable,
952 def patch_all_gather(new_all_gather_into_tensor: Callable):
964 def patch_reduce_scatter(new_reduce_scatter_tensor: Callable):
976 def patch_all_reduce(new_all_reduce: Callable):
989 def patch_unshard(new_unshard: Callable):
1002 def patch_reshard(new_reshard: Callable):
1015 def patch_post_backward(new_post_backward: Callable):
1028 def patch_register_post_backward_hook_backward(new_backward: Callable):
1041 orig_reduce_scatter: Callable,
1042 assert_fn: Callable, # `assert_fn(output: Tensor)`
[all …]

12345678910>>...37