Home
last modified time | relevance | path

Searched +defs:func +defs:types (Results 1 – 25 of 613) sorted by relevance

12345678910>>...25

/aosp_15_r20/external/pytorch/test/
H A Dtest_python_dispatch.py915 def __torch_dispatch__(cls, func, types, args=(), kwargs=None): argument
935 def __torch_dispatch__(cls, func, types, args=(), kwargs=None): argument
1004 def __torch_dispatch__(cls, func, types, args=(), kwargs=None): argument
1013 def __torch_dispatch__(cls, func, types, args=(), kwargs=None): argument
1109 def __torch_dispatch__(cls, func, types, args=(), kwargs=None): argument
1117 def __torch_dispatch__(cls, func, types, args=(), kwargs=None): argument
1191 def __torch_dispatch__(cls, func, types, args=(), kwargs=None): argument
1247 def __torch_dispatch__(cls, func, types, args=(), kwargs=None): argument
1318 def __torch_dispatch__(cls, func, types, args=(), kwargs=None): argument
1365 def __torch_dispatch__(cls, func, types, args=(), kwargs=None): argument
[all …]
H A Dtest_overrides.py86 def decorator(func): argument
145 def __torch_function__(cls, func, types, args=(), kwargs=None): argument
189 def decorator(func): argument
215 def __torch_function__(cls, func, types, args=(), kwargs=None): argument
254 def decorator(func): argument
318 def decorator(func): argument
366 def __torch_function__(cls, func, types, args=(), kwargs=None): argument
600 def __torch_function__(cls, func, types, args=(), kwargs=None): argument
612 def __torch_function__(cls, func, types, args=(), kwargs=None): argument
617 def __torch_function__(cls, func, types, args=(), kwargs=None): argument
[all …]
H A Dtest_autocast.py216 def __torch_dispatch__(self, func, types, args=(), kwargs=None): argument
325 def __torch_dispatch__(self, func, types, args=(), kwargs=None): argument
/aosp_15_r20/external/pytorch/torch/sparse/
H A D_semi_structured_ops.py29 def fallback_dispatcher(func, types, args, kwargs): argument
34 def semi_sparse_values(func, types, args=(), kwargs=None) -> torch.Tensor: argument
47 def semi_sparse_indices(func, types, args=(), kwargs=None) -> torch.Tensor: argument
61 def semi_sparse_t(func, types, args=(), kwargs=None) -> torch.Tensor: argument
83 def semi_sparse_view(func, types, args=(), kwargs=None) -> torch.Tensor: argument
93 def semi_sparse_detach(func, types, args, kwargs) -> torch.Tensor: argument
107 def semi_sparse_mm(func, types, args=(), kwargs=None) -> torch.Tensor: argument
128 def semi_sparse_addmm(func, types, args=(), kwargs=None) -> torch.Tensor: argument
151 def semi_sparse_linear(func, types, args=(), kwargs=None) -> torch.Tensor: argument
/aosp_15_r20/external/jazzer-api/src/main/java/com/code_intelligence/jazzer/autofuzz/
H A DMeta.java79 public static <T1> void autofuzz(FuzzedDataProvider data, Consumer1<T1> func) { in autofuzz()
80 Class<?>[] types = TypeResolver.resolveRawArguments(Consumer1.class, func.getClass()); in autofuzz() local
85 public static <T1, T2> void autofuzz(FuzzedDataProvider data, Consumer2<T1, T2> func) { in autofuzz()
86 Class<?>[] types = TypeResolver.resolveRawArguments(Consumer2.class, func.getClass()); in autofuzz() local
92 public static <T1, T2, T3> void autofuzz(FuzzedDataProvider data, Consumer3<T1, T2, T3> func) { in autofuzz()
93 Class<?>[] types = TypeResolver.resolveRawArguments(Consumer3.class, func.getClass()); in autofuzz() local
101 FuzzedDataProvider data, Consumer4<T1, T2, T3, T4> func) { in autofuzz()
102 Class<?>[] types = TypeResolver.resolveRawArguments(Consumer4.class, func.getClass()); in autofuzz() local
111 FuzzedDataProvider data, Consumer5<T1, T2, T3, T4, T5> func) { in autofuzz()
112 Class<?>[] types = TypeResolver.resolveRawArguments(Consumer5.class, func.getClass()); in autofuzz() local
[all …]
/aosp_15_r20/external/pytorch/test/dynamo/
H A Dtest_subclasses.py151 def __torch_function__(cls, func, types, args=(), kwargs=None): argument
159 def __torch_function__(cls, func, types, args=(), kwargs=None): argument
170 def __torch_function__(cls, func, types, args=(), kwargs=None): argument
179 def __torch_function__(cls, func, types, args=(), kwargs=None): argument
194 def __torch_function__(cls, func, types, args=(), kwargs=None): argument
206 def __torch_function__(cls, func, types, args=(), kwargs=None): argument
256 def __torch_dispatch__(cls, func, types, args, kwargs=None): argument
305 def __torch_dispatch__(cls, func, types, args, kwargs=None): argument
356 def __torch_dispatch__(cls, func, types, args, kwargs): argument
388 def func(a): function
[all …]
/aosp_15_r20/external/pytorch/torch/fx/experimental/unification/multipledispatch/
H A Ddispatcher.py49 def variadic_signature_matches_iter(types, full_signature): argument
88 def variadic_signature_matches(types, full_signature): argument
121 def register(self, *types, **kwargs):
142 def _df(func): argument
148 def get_func_params(cls, func): argument
154 def get_func_annotations(cls, func): argument
173 def add(self, signature, func): argument
280 def dispatch(self, *types):
306 def dispatch_iter(self, *types):
319 def resolve(self, types): argument
[all …]
H A Dcore.py11 def dispatch(*types, **kwargs):
53 def _df(func): argument
71 def ismethod(func): argument
/aosp_15_r20/external/pytorch/test/nn/
H A Dtest_load_state_dict.py456 def load_torch_function_handler(cls, func, types, args=(), kwargs=None): argument
507 def __torch_function__(cls, func, types, args=(), kwargs=None): argument
515 def __torch_function__(cls, func, types, args=(), kwargs=None): argument
521 def __torch_function__(cls, func, types, args=(), kwargs=None): argument
557 def __torch_dispatch__(cls, func, types, args=(), kwargs=None): argument
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dcommon_subclass.py70 def __torch_dispatch__(cls, func, types, args=(), kwargs=None): argument
125 def __torch_dispatch__(cls, func, types, args=(), kwargs=None): argument
152 def _try_call_special_impl(cls, func, args, kwargs): argument
168 def __torch_function__(cls, func, types, args=(), kwargs=None): argument
251 def __torch_dispatch__(cls, func, types, args, kwargs): argument
H A Dcomposite_compliance.py43 def is_view_fn(func): argument
83 def is_inplace_view_fn(func): argument
97 def is_inplace(func): argument
162 def __torch_dispatch__(cls, func, types, args=(), kwargs=None): argument
171 def __torch_dispatch__(self, func, types, args=(), kwargs=None): argument
H A Dtwo_tensor.py53 def __torch_dispatch__(cls, func, types, args, kwargs): argument
82 def __torch_dispatch__(self, func, types, args=(), kwargs=None): argument
H A Dlogging_tensor.py74 def __torch_dispatch__(cls, func, types, args=(), kwargs=None): argument
87 def __torch_dispatch__(self, func, types, args=(), kwargs=None): argument
/aosp_15_r20/external/pytorch/benchmarks/overrides_benchmark/
H A Dcommon.py21 def __torch_function__(cls, func, types, args=(), kwargs=None): argument
30 def __torch_function__(cls, func, types, args=(), kwargs=None): argument
/aosp_15_r20/external/pytorch/torch/
H A Dlibrary.py502 def impl(qualname, types, func=None, *, lib=None): argument
537 def _impl(qualname, types, func=None, *, lib=None, disable_dynamo=False): argument
554 def register(func): argument
606 def impl_abstract(qualname, func=None, *, lib=None, _stacklevel=1): argument
788 def register(func): argument
996 def register(func): argument
1117 def register(func): argument
1145 def _check_pystubs_once(func, qualname, actual_module_name): argument
H A Doverrides.py1574 def inner(func): argument
2032 def __torch_function__(self, func, types, args=(), kwargs=None): argument
2080 def __torch_function__(self, func, types, args=(), kwargs=None): argument
/aosp_15_r20/external/pytorch/test/export/
H A Dopinfo_schema.py39 def _may_alias_or_mutate(self, func, types, args, kwargs): argument
74 def __torch_dispatch__(self, func, types, args=(), kwargs=None): argument
/aosp_15_r20/external/pytorch/torch/utils/
H A D_python_dispatch.py82 def __torch_dispatch__(self, func, types, args=(), kwargs=None): argument
288 def __torch_dispatch__(self, func, types, args=(), kwargs=None): argument
440 def _correct_storage_aliasing(func, schema_info, args, outs): argument
525 def get_alias_info(func) -> SchemaInfo: argument
592 def return_and_correct_aliasing(func, args, kwargs, out): argument
/aosp_15_r20/external/tensorflow/tensorflow/python/util/
H A Ddispatch.py181 def __init__(self, override_func, types): argument
201 def dispatch_for_types(op, *types):
220 def decorator(func): argument
424 def _add_name_scope_wrapper(func, api_signature): argument
569 def _check_signature(api_signature, func): argument
691 def _signature_from_annotations(func): argument
921 def register_unary_elementwise_api(func): argument
930 def register_binary_elementwise_api(func): argument
939 def register_binary_elementwise_assert_api(func): argument
/aosp_15_r20/external/pytorch/test/quantization/core/experimental/
H A Dtest_bits.py21 def __torch_dispatch__(cls, func, types, args=(), kwargs=None): argument
44 def __torch_function__(cls, func, types, args=(), kwargs=None): argument
/aosp_15_r20/external/pytorch/torch/nested/_internal/
H A Dnested_tensor.py35 def _get_sdpa_extreme_seqlen(func, tensor): argument
285 def __torch_dispatch__(cls, func, types, args=(), kwargs=None): argument
298 def __torch_function__(cls, func, types, args=(), kwargs=None): argument
/aosp_15_r20/external/tensorflow/tensorflow/core/transforms/functional_to_region/
H A Dimpl.cc168 static bool CannotInline(GraphFuncOp func) { in CannotInline()
186 static DictionaryAttr PreserveFunctionAttributes(GraphFuncOp func) { in PreserveFunctionAttributes()
242 static RegionAttr PreserveAttributes(GraphFuncOp func, bool drop_args = false, in PreserveAttributes()
266 YieldOp BasePattern::ReplaceReturnWithYield(Block &block, TypeRange types, in ReplaceReturnWithYield()
275 void BasePattern::CloneAndReorderArgs(TypeRange types, Region &from, Region &to, in CloneAndReorderArgs()
379 for (GraphFuncOp func : branch_funcs) { in matchAndRewrite() local
/aosp_15_r20/external/pytorch/test/profiler/
H A Dtest_profiler_tree.py56 def __torch_function__(cls, func, types, args=(), kwargs=None): argument
68 def __torch_dispatch__(cls, func, types, args=(), kwargs=None): argument
/aosp_15_r20/external/aws-sdk-java-v2/codegen/src/main/java/software/amazon/awssdk/codegen/poet/rules2/
H A DRuleRuntimeTypeMirror.java61 Map<String, RuleType> types = new HashMap<>(); in RuleRuntimeTypeMirror() local
110 static Map<String, RuleFunctionMirror> buildNameToFunctionMap(Map<String, RuleType> types) { in buildNameToFunctionMap()
113 result.put(func.name(), func); in buildNameToFunctionMap() local
118 static List<RuleFunctionMirror> builtInFunctions(Map<String, RuleType> types) { in builtInFunctions()
/aosp_15_r20/external/pytorch/torch/csrc/utils/
H A Ddisable_torch_function.cpp209 PyObject *func = nullptr, *types = nullptr, *args = nullptr, in THPModule_disable_torch_function() local
243 PyObject *func = nullptr, *types = nullptr, *args = nullptr, in THPModule_disable_torch_dispatch() local

12345678910>>...25