/aosp_15_r20/external/pytorch/.github/scripts/ |
H A 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 …]
|
H A 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 …]
|
H A 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 …]
|
/aosp_15_r20/external/pytorch/torch/testing/ |
H A 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 …]
|
/aosp_15_r20/external/pytorch/torch/ao/quantization/ |
H A 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 …]
|
/aosp_15_r20/external/pytorch/test/jit/ |
H A 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 …]
|
/aosp_15_r20/external/pytorch/torch/utils/ |
H A D | _pytree.py | 109 def get(self, parent: Any) -> Any: 172 cls: Type[Any], 244 cls: Type[Any], 277 cls: Type[Any], 332 cls: Type[Any], 407 def get(self, obj: Any) -> Any: 411 def _tuple_flatten(d: Tuple[Any, ...]) -> Tuple[List[Any], Context]: 416 d: Tuple[Any, ...] 422 def _tuple_unflatten(values: Iterable[Any], context: Context) -> Tuple[Any, ...]: 426 def _list_flatten(d: List[Any]) -> Tuple[List[Any], Context]: [all …]
|
H A D | _cxx_pytree.py | 95 def wrapped(*args: Any, **kwargs: Any) -> Any: 102 cls: Type[Any], 172 cls: Type[Any], 223 cls: Type[Any], 294 def tree_unflatten(leaves: Iterable[Any], treespec: TreeSpec) -> PyTree: 427 func: Callable[..., Any], argument 476 func: Callable[..., Any], argument 552 def map_only(__type_or_types_or_pred: Callable[[Any], bool]) -> MapOnlyFn[FnAny[Any]]: 557 __type_or_types_or_pred: Union[TypeAny, Callable[[Any], bool]] 582 def pred(x: Any) -> bool: [all …]
|
/aosp_15_r20/external/golang-protobuf/types/known/anypb/ |
H A 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 …]
|
/aosp_15_r20/external/pytorch/torch/_inductor/codegen/ |
H A 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 …]
|
/aosp_15_r20/external/pytorch/torch/export/ |
H A D | _trace.py | 351 flat_fake_args: List[Any], 522 args: Tuple[Any, ...], 523 kwargs: Optional[Dict[str, Any]] = None, argument 524 dynamic_shapes: Optional[Union[Dict[str, Any], Tuple[Any], List[Any]]] = None, 796 args: Tuple[Any, ...], 797 kwargs: Optional[Dict[str, Any]] = None, argument 1105 export_artifact: ExportArtifact, combined_args: Dict[str, Any], dynamic_shapes argument 1213 args: Tuple[Any, ...], 1214 kwargs: Dict[str, Any], argument 1215 dynamic_shapes: Optional[Union[Dict[str, Any], Tuple[Any], List[Any]]], [all …]
|
H A 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],
|
/aosp_15_r20/external/pytorch/torch/onnx/_internal/ |
H A 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 …]
|
/aosp_15_r20/external/pytorch/torch/distributed/ |
H A 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 …]
|
/aosp_15_r20/external/pytorch/torch/_inductor/ |
H A D | pattern_matcher.py | 110 def __call__(self, *args: Any, **kwargs: Any) -> Any: 115 def __call__(self, *args: Any, **kwargs: Any) -> Any: 121 self, fn: Union[SearchFn, ReplaceFn], *args: Any, **kwargs: Any 166 args: Optional[Sequence[Any]] = None, 219 self, replacement_graph: torch.fx.Graph, args: Sequence[Any] 228 args: Sequence[Any], 274 def __init__(self, format_string: str, *args: Any, **kwargs: Any) -> None: 372 def pattern_eq(self, other: Any) -> bool: 421 def pattern_eq(self, other: Any) -> bool: 447 def pattern_eq(self, other: Any) -> bool: [all …]
|
/aosp_15_r20/external/pytorch/tools/lldb/ |
H A 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:
|
/aosp_15_r20/external/pytorch/torch/autograd/ |
H A 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:
|
/aosp_15_r20/external/libbrillo/brillo/ |
H A 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 …]
|
/aosp_15_r20/external/pytorch/torch/distributed/fsdp/ |
H A D | _optim_utils.py | 74 def sorted_items(dictionary: Dict[str, Any]) -> Iterator[Tuple[str, Any]]: argument 135 flat_param_state: Dict[str, Any], argument 188 def _is_zero_dim_tensor(x: Any) -> bool: 194 flat_param_state: Dict[str, Any], argument 341 optim_state: Dict[str, Any], argument 359 fsdp_state: _FSDPState, state: Any, group: Optional[dist.ProcessGroup] 384 optim_state: Dict[str, Any], argument 417 optim_state_dict: Dict[str, Any], argument 563 unflat_osd_state: Dict[str, Dict[str, Any]], argument 840 non_tensors: List[Any], [all …]
|
/aosp_15_r20/external/python/absl-py/absl/flags/ |
D | _defines.py | 63 default: Any, 69 **args: Any 78 default: Optional[Any], 84 **args: Any 318 module: Any, flag_values: _flagvalues.FlagValues = _flagvalues.FLAGS 378 **args: Any 390 **args: Any 402 **args: Any 438 **args: Any 451 **args: Any [all …]
|
/aosp_15_r20/external/pytorch/tools/test/ |
H A 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 …]
|
/aosp_15_r20/external/pytorch/torch/_export/ |
H A 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],
|
/aosp_15_r20/external/pytorch/tools/flight_recorder/components/ |
H A 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 …]
|
/aosp_15_r20/external/executorch/.github/scripts/ |
H A D | github_utils.py | 38 data: Union[Optional[Dict[str, Any]], str] = None, argument 40 reader: Callable[[Any], Any] = lambda x: x.read(), 73 data: Union[Optional[Dict[str, Any]], str] = None, argument 75 reader: Callable[[Any], Any] = json.load, 84 params: Optional[Dict[str, Any]] = None, argument 101 params: Optional[Dict[str, Any]] = None, argument 114 params: Optional[Dict[str, Any]] = None, argument 122 params: Optional[Dict[str, Any]] = None, argument 128 def gh_graphql(query: str, **kwargs: Any) -> Dict[str, Any]:
|
/aosp_15_r20/external/pytorch/torchgen/_autoheuristic/ |
H A D | ah_tree.py | 14 class_probs: Any = None, 37 def __init__(self, sklearn_tree: Any, feature_names: List[str]) -> None: 43 self, sklearn_tree: Any, node_id: int = 0 70 def prune(self, df: Any, target_col: str, k: int) -> None: 74 self, node: DecisionTreeNode, df: Any, target_col: str, k: int 146 self, class_probs: Any, num_samples: int, max_per_line: int = 5 159 def predict(self, X: Any) -> Any: 163 def predict_proba(self, X: Any) -> Any: 166 def _get_leaf(self, X: Any) -> DecisionTreeNode: 177 def _predict_single(self, x: Any) -> str: [all …]
|