| /external/pytorch/torch/fx/passes/ |
| D | pass_manager.py | 20 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):
|
| /external/pytorch/torch/fx/passes/infra/ |
| D | pass_manager.py | 19 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:
|
| /external/pytorch/torch/ |
| D | _vmap_internals.py | 44 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,
|
| /external/cronet/tot/third_party/llvm-libc/src/test/UnitTest/ |
| D | FPExceptMatcher.h | 34 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()
|
| /external/cronet/stable/third_party/llvm-libc/src/test/UnitTest/ |
| D | FPExceptMatcher.h | 34 struct Callable : public FunctionCaller { in getFunctionCaller() struct
35 Func f; in getFunctionCaller()
36 explicit Callable(Func theFunc) : f(theFunc) {} in getFunctionCaller() argument
37 void call() override { f(); } in getFunctionCaller()
|
| /external/llvm-libc/test/UnitTest/ |
| D | FPExceptMatcher.h | 34 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()
|
| /external/pytorch/torch/utils/data/_utils/ |
| D | collate.py | 121 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
|
| /external/pytorch/torch/utils/benchmark/utils/ |
| D | compile.py | 41 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
|
| /external/pytorch/torch/_functorch/ |
| D | deprecated.py | 59 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:
|
| D | aot_autograd.py | 769 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 …]
|
| D | apis.py | 33 func: Callable,
214 func: Callable,
300 def grad(func: Callable, argnums: argnums_t = 0, has_aux: bool = False) -> Callable:
409 func: Callable, argnums: argnums_t = 0, has_aux: bool = False
|
| /external/pytorch/torch/fx/ |
| D | subgraph_rewriter.py | 85 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],
|
| /external/pytorch/torch/ao/ns/ |
| D | _numeric_suite_fx.py | 313 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 …]
|
| /external/executorch/exir/ |
| D | control_flow.py | 96 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]]
|
| /external/crosvm/tools/impl/ |
| D | cli.py | 38 def run_main(main_fn: Callable[..., Any], usage: Optional[str] = None):
43 *functions: Callable[..., Any],
44 default_fn: Optional[Callable[..., Any]] = None,
|
| /external/python/pyee/pyee/ |
| D | base.py | 111 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:
|
| /external/grpc-grpc/src/python/grpcio/grpc/ |
| D | _common.py | 112 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,
|
| /external/pytorch/torch/onnx/_internal/diagnostics/infra/ |
| D | utils.py | 43 def _function_source_info(fn: Callable) -> tuple[Sequence[str], int, str | None]:
53 def function_location(fn: Callable) -> _infra.Location:
66 fn: Callable, args: tuple[Any, ...], kwargs: dict[str, Any]
|
| /external/pytorch/torch/_library/ |
| D | custom_ops.py | 35 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,
|
| /external/pytorch/torch/ao/quantization/ |
| D | quantization_mappings.py | 239 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:
|
| /external/pytorch/torch/utils/data/datapipes/iter/ |
| D | callable.py | 72 fn: Callable,
227 Callable[..., Any], Dict[Union[str, Any], Union[Callable, Any]], None
229 collate_fn: Optional[Callable] = None,
|
| D | routeddecoder.py | 42 *handlers: Callable,
43 key_fn: Callable = extension_extract_fn,
57 def add_handler(self, *handler: Callable) -> None:
|
| /external/pigweed/pw_rpc/py/pw_rpc/console_tools/ |
| D | watchdog.py | 29 on_reset: Callable[[], Any],
30 on_expiration: Callable[[], Any],
31 while_expired: Callable[[], Any] = lambda: None,
|
| /external/pigweed/pw_console/py/pw_console/widgets/ |
| D | border.py | 33 title: Callable[[], str | StyleAndTextTuples] | str = '',
34 border_style: Callable[[], str] | str = '',
35 base_style: Callable[[], str] | str = '',
|
| /external/pytorch/torch/_higher_order_ops/ |
| D | flex_attention.py | 97 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 …]
|