| /external/pytorch/.github/scripts/ |
| D | test_tryrebase.py | 25 mocked_post_comment: Any,
26 mocked_rp: Any,
27 mocked_run_git: Any,
28 mocked_gql: Any,
56 mocked_post_comment: Any,
57 mocked_rp: Any,
58 mocked_run_git: Any,
59 mocked_gql: Any,
87 mocked_post_comment: Any,
88 mocked_rp: Any,
[all …]
|
| D | test_check_labels.py | 64 self, mock_gh_post_pr_comment: Any, mock_get_comments: Any, mock_gh_grphql: Any
75 self, mock_gh_post_pr_comment: Any, mock_get_comments: Any, mock_gh_grphql: Any
86 self, mock_gh_delete_comment: Any, mock_get_comments: Any, mock_gh_grphql: Any
105 mock_add_label_err_comment: Any,
106 mock_delete_all_label_err_comments: Any,
107 mock_has_required_labels: Any,
108 mock_parse_args: Any,
109 mock_gh_get_info: Any,
129 mock_add_label_err_comment: Any,
130 mock_delete_all_label_err_comments: Any,
[all …]
|
| D | test_trymerge.py | 50 fallback_function: Any,
52 key_function: Any,
53 *args: Any,
63 def save_mocked_queries(obj: Any) -> None:
101 def mocked_gh_graphql(query: str, **kwargs: Any) -> Any:
102 def key_function(query: str, kwargs: Any) -> str:
107 def gh_graphql_wrapper(query: str, kwargs: Any) -> Any:
190 def mocked_read_merge_rules_NE(repo: Any, org: str, project: str) -> List[MergeRule]:
202 def mocked_read_merge_rules(repo: Any, org: str, project: str) -> List[MergeRule]:
230 repo: Any, org: str, project: str
[all …]
|
| D | test_label_utils.py | 42 mock_request_for_labels: Any,
43 mock_get_last_page_num_from_header: Any,
53 mock_request_for_labels: Any,
54 get_last_page_num_from_header: Any,
65 self, mocked_rn_labels: Any, mocked_gql: Any
76 self, mocked_rn_labels: Any, mocked_gql: Any
87 self, mocked_rn_labels: Any, mocked_gql: Any
|
| /external/pytorch/tools/lldb/ |
| D | pytorch_lldb.py | 27 def __exit__(self, etype: Any, evalue: Any, tb: Any) -> None:
34 def IntArrayRef_summary(valobj: Any, internal_dict: Any, options: Any) -> str:
47 def DispatchKeyset_summary(valobj: Any, internal_dict: Any, options: Any) -> str:
60 def Tensor_summary(valobj: Any, internal_dict: Any, options: Any) -> str:
81 def __lldb_init_module(debugger: Any, internal_dict: Any) -> Any:
|
| /external/golang-protobuf/types/known/anypb/ |
| D | any.pb.go | 212 type Any struct { struct
213 state protoimpl.MessageState
214 sizeCache protoimpl.SizeCache
215 unknownFields protoimpl.UnknownFields
244 TypeUrl string `protobuf:"bytes,1,opt,name=type_url,json=typeUrl,proto3" json:"type_url,omitempty"`
246 Value []byte `protobuf:"bytes,2,opt,name=value,proto3" json:"value,omitempty"`
262 func MarshalFrom(dst *Any, src proto.Message, opts proto.MarshalOptions) error {
281 func UnmarshalTo(src *Any, dst proto.Message, opts proto.UnmarshalOptions) error {
300 func UnmarshalNew(src *Any, opts proto.UnmarshalOptions) (dst proto.Message, err error) {
323 func (x *Any) MessageIs(m proto.Message) bool {
[all …]
|
| /external/pytorch/test/jit/ |
| D | test_isinstance.py | 28 def int_test(x: Any):
36 def float_test(x: Any):
44 def bool_test(x: Any):
52 def list_str_test(x: Any):
61 def list_tensor_test(x: Any):
69 def dict_str_int_test(x: Any):
78 def dict_int_tensor_test(x: Any):
85 def tuple_test(x: Any):
94 def tuple_tensor_test(x: Any):
101 def optional_test(x: Any):
[all …]
|
| /external/pytorch/torch/ao/quantization/ |
| D | quantize_fx.py | 25 preserved_attrs: Dict[str, Any], argument
76 fuse_custom_config: Union[FuseCustomConfig, Dict[str, Any], None] = None, argument
92 qconfig_mapping: Union[QConfigMapping, Dict[str, Any]], argument
94 example_inputs: Tuple[Any, ...],
96 _equalization_config: Optional[Union[QConfigMapping, Dict[str, Any]]] = None, argument
164 qconfig_mapping: Union[QConfigMapping, Dict[str, Any]], argument
166 example_inputs: Tuple[Any, ...],
206 fuse_custom_config: Union[FuseCustomConfig, Dict[str, Any], None] = None, argument
254 qconfig_mapping: Union[QConfigMapping, Dict[str, Any]], argument
255 example_inputs: Tuple[Any, ...],
[all …]
|
| /external/pytorch/torch/testing/ |
| D | _comparison.py | 37 self, type: Type[Exception], msg: str, *, id: Tuple[Any, ...] = ()
114 id: Tuple[Any, ...] = (),
329 actual: Any,
330 expected: Any,
332 id: Tuple[Any, ...] = (),
333 **unknown_parameters: Any,
345 def _check_inputs_isinstance(*inputs: Any, cls: Union[Type, Tuple[Type, ...]]):
351 self, type: Type[Exception], msg: str, *, id: Tuple[Any, ...] = ()
421 def __init__(self, actual: Any, expected: Any, **other_parameters: Any) -> None:
445 actual: Any,
[all …]
|
| /external/pytorch/torch/export/ |
| D | __init__.py | 75 args: Tuple[Any, ...],
76 kwargs: Optional[Dict[str, Any]] = None, argument
78 dynamic_shapes: Optional[Union[Dict[str, Any], Tuple[Any], List[Any]]] = None,
166 args: Tuple[Any, ...],
167 kwargs: Optional[Dict[str, Any]] = None, argument
169 dynamic_shapes: Optional[Union[Dict[str, Any], Tuple[Any], List[Any]]] = None,
285 extra_files: Optional[Dict[str, Any]] = None, argument
366 extra_files: Optional[Dict[str, Any]] = None, argument
479 cls: Type[Any],
|
| /external/pytorch/tools/test/ |
| D | test_upload_stats_lib.py | 57 def test_emits_default_and_given_metrics(self, mock_resource: Any) -> None:
90 def mock_put_item(Item: dict[str, Any]) -> None: argument
105 self, mock_resource: Any
124 def mock_put_item(Item: dict[str, Any]) -> None: argument
139 self, mock_resource: Any
160 def mock_put_item(Item: dict[str, Any]) -> None: argument
175 self, mock_resource: Any
196 def mock_put_item(Item: dict[str, Any]) -> None: argument
211 self, mock_resource: Any
229 def mock_put_item(Item: dict[str, Any]) -> None: argument
[all …]
|
| /external/python/bumble/bumble/pandora/ |
| D | utils.py | 45 self, msg: str, kwargs: MutableMapping[str, Any] argument
77 def rpc(func: Any) -> Any:
80 self: Any, request: Any, context: grpc.ServicerContext
88 self: Any, request: Any, context: grpc.ServicerContext
94 def gen_wrapper(self: Any, request: Any, context: grpc.ServicerContext) -> Any:
100 def wrapper(self: Any, request: Any, context: grpc.ServicerContext) -> Any:
|
| /external/pytorch/tools/flight_recorder/components/ |
| D | builder.py | 42 def tabulate(data: Any, headers: Any = None) -> Any: # type: ignore[misc]
52 pg_config: Any,
122 entry: Dict[Any, Any], argument
124 collective_id: Any,
126 global_rank: Any,
140 all_entries: Dict[int, List[Dict[str, Any]]], argument
142 _memberships: Dict[str, Set[Any]],
416 details: Dict[str, Dict[str, Any]], args: argparse.Namespace, version: str argument
|
| D | utils.py | 41 event_a: Dict[Any, Any], argument
42 event_b: Dict[Any, Any],
43 memberships: Dict[str, Set[Any]],
52 all_rank_events: Dict[Any, Any], argument
55 memberships: Dict[str, Set[Any]],
172 def check_size_alltoall(alltoall_cases: List[Dict[str, Any]]) -> Tuple[bool, int, int]: argument
183 entries: List[Dict[str, Any]], argument
214 all_entries: Dict[int, List[Dict[str, Any]]], argument
216 _memberships: Dict[str, Set[Any]],
254 entries: Dict[int, Any], memberships: List[Membership] argument
[all …]
|
| /external/executorch/exir/_serialize/ |
| D | _dataclass.py | 17 def default(self, o: Any) -> Any:
35 def _is_optional(T: Any) -> bool:
44 def _is_strict_union(T: Any, cls: Any, key: str) -> bool:
49 def _get_class_from_union(json_dict: Dict[str, Any], key: str, cls: Any) -> Any: argument
60 def _json_to_dataclass(json_dict: Dict[str, Any], cls: Any = None) -> Any: argument
|
| /external/pytorch/torch/_inductor/runtime/ |
| D | benchmarking.py | 22 fn: Callable[Concatenate[Any, P], T]
35 def wrapper(self: Any, *args: P.args, **kwargs: P.kwargs) -> T:
51 def count(fn: Callable[Concatenate[Any, P], T]) -> Callable[Concatenate[Any, P], T]:
62 def wrapper(self: Any, *args: P.args, **kwargs: P.kwargs) -> T:
79 fn: Callable[..., Any], argument
80 fn_args: Tuple[Any],
82 **kwargs: Any,
128 self: Self, _callable: Callable[[], Any], warmup: int = 20, rep: int = 100 argument
162 def benchmark_gpu(self: Self, *args: Any, **kwargs: Any) -> float:
180 def benchmark_gpu(self: Self, _callable: Callable[[], Any], **kwargs: Any) -> float: argument
|
| /external/pytorch/torch/distributed/ |
| D | _state_dict_utils.py | 38 companion_obj: Any,
86 iter_object: Any,
94 companion_obj: Any = None,
207 state_dict: Dict[str, Any], argument
295 state_dict: Dict[str, Any], argument
335 state_dict: Dict[str, Any], argument
384 state_dict: Dict[str, Any], pin_memory: bool = False, share_memory: bool = False argument
405 _: Any,
453 state_dict: Dict[str, Any], argument
469 companion_obj: Any,
[all …]
|
| /external/pytorch/torch/onnx/_internal/ |
| D | io_adapter.py | 40 model_args: Sequence[Any],
98 model_outputs: Any,
119 model_outputs: Any,
188 model_args: Sequence[Any],
225 model_args: Sequence[Any],
250 model_args: Sequence[Any],
278 model_args: Sequence[Any],
312 model_args: Sequence[Any],
371 model_args: Sequence[Any],
413 model_args: Sequence[Any],
[all …]
|
| /external/pytorch/torch/autograd/ |
| D | grad_mode.py | 84 def __exit__(self, exc_type: Any, exc_value: Any, traceback: Any) -> None:
139 def __exit__(self, exc_type: Any, exc_value: Any, traceback: Any) -> None:
196 def __exit__(self, exc_type: Any, exc_value: Any, traceback: Any) -> None:
273 def __exit__(self, exc_type: Any, exc_value: Any, traceback: Any) -> None:
319 def __exit__(self, exc_type: Any, exc_value: Any, traceback: Any) -> None:
359 def __exit__(self, exc_type: Any, exc_value: Any, traceback: Any) -> None:
|
| /external/pandora/avatar/avatar/ |
| D | __init__.py | 177 def __init__(self, func: Callable[..., Any]) -> None: argument
182 def parameterized(*inputs: Tuple[Any, ...]) -> Type[Wrapper]:
187 def decorate(input: Tuple[Any, ...]) -> Callable[..., Any]:
189 def wrapper(*args: Any, **kwargs: Any) -> Any:
194 def normalize(a: Any) -> Any:
224 def getFlagValue(server: PandoraServer[Any], flag: str) -> str:
233 def isFlagEnabled(server: PandoraServer[Any], flag: str) -> bool:
237 def isFlagValidForTest(server: PandoraServer[Any], flag: str) -> bool:
240 def decorator(func: Callable[..., Any]) -> Callable[..., Any]: argument
242 def wrapper(self: base_test.BaseTestClass, *args: Any, **kwargs: Any) -> Any:
[all …]
|
| /external/pytorch/torch/nn/parallel/ |
| D | parallel_apply.py | 10 def get_a_var(obj: Union[torch.Tensor, List[Any], Tuple[Any, ...], Dict[Any, Any]]) -> Optional[tor… argument
26 inputs: Sequence[Any],
27 kwargs_tup: Optional[Sequence[Dict[str, Any]]] = None, argument
60 input: Any,
61 kwargs: Dict[str, Any], argument
|
| /external/pytorch/torch/_inductor/codegen/ |
| D | triton_combo_kernel.py | 53 node_info_map: Dict[BaseSchedulerNode, Tuple[Any, Any, Any, Any]], argument
143 Dict[BaseSchedulerNode, Tuple[Any, Any, Any, Any]], argument
192 node_info_map: Dict[BaseSchedulerNode, Tuple[Any, Any, Any, Any]], argument
248 node_info_map: Dict[BaseSchedulerNode, Tuple[Any, Any, Any, Any]], argument
468 *groups: Any,
664 signature: Optional[List[Any]] = None,
762 def add_numel_to_args(self, argdefs: List[str], signature: List[Any]) -> List[str]:
774 self, name: str, call_args: List[Any], arg_types: List[Any], grid: List[Any]
802 self, extra_args: List[Any], grid: Union[List[Any], Tuple[Any, ...]]
887 self, num_gb: float, grid: Optional[List[Any]] = None
[all …]
|
| /external/pytorch/torch/_export/ |
| D | non_strict_utils.py | 86 t: Any,
230 combined_args: Dict[str, Any], argument
231 dynamic_shapes: Union[Dict[str, Any], Tuple[Any], List[Any]],
246 dynamic_shapes: Union[Dict[str, Any], Tuple[Any], List[Any], None], argument
311 combined_args: Dict[str, Any], argument
312 dynamic_shapes: Union[Dict[str, Any], Tuple[Any], List[Any], None],
419 args: Tuple[Any],
420 kwargs: Dict[Any, Any],
|
| /external/libbrillo/brillo/ |
| D | any_test.cc | 16 TEST(Any, Empty) { in TEST() argument
29 TEST(Any, SimpleTypes) { in TEST() argument
46 TEST(Any, Clear) { in TEST() argument
55 TEST(Any, Assignments) { in TEST() argument
98 TEST(Any, Enums) { in TEST() argument
113 TEST(Any, Integers) { in TEST() argument
142 TEST(Any, Pointers) { in TEST() argument
161 TEST(Any, Arrays) { in TEST() argument
187 TEST(Any, References) { in TEST() argument
210 TEST(Any, CustomTypes) { in TEST() argument
[all …]
|
| /external/pytorch/torchgen/_autoheuristic/pad_mm/ |
| D | gen_data_pad_mm.py | 59 dtype: Any,
80 def mm(a: Any, b: Any) -> Any:
83 def mm_mat1_prepadded(a: Any, b: Any) -> Any:
86 def mm_mat2_prepadded(a: Any, b: Any) -> Any:
89 def mm_mat1_mat2_prepadded(a: Any, b: Any) -> Any:
116 def get_m_k_n(self, dtype: Any) -> Tuple[int, int, int]:
|