Home
last modified time | relevance | path

Searched full:category_override (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/pytorch/torchgen/api/
H A Dpython.py746 f.func, category_override=f.category_override, method=method, pyi=pyi
753 category_override: str | None,
800 is_factory_function = category_override == "factory" or (
804 category_override in ("new", "like")
808 is_dummy_function = category_override == "dummy"
/aosp_15_r20/external/pytorch/torchgen/
H A Dmodel.py501 category_override: str | None
688 category_override = e.pop("category_override", None)
689 assert category_override is None or isinstance(
690 category_override, str
691 ), f"not a str: {category_override}"
939 category_override=category_override,
H A Dgen.py1266 "category_override",
1267 f.category_override if f.category_override is not None else "",
H A Dnative_function_generation.py352 category_override=None,
/aosp_15_r20/external/pytorch/tools/autograd/
H A Dgen_python_functions.py661 category_override=pair.function.category_override,
H A Dgen_trace_type.py192 is_factory_method = f.category_override == "factory" or (
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A Dnative_functions.yaml2431 category_override: factory
2463 category_override: factory
3407 category_override: factory
3412 category_override: factory
3417 category_override: factory
3428 category_override: factory
3433 category_override: factory
3438 category_override: factory
3623 category_override: factory
3628 category_override: factory
[all …]
H A DREADME.md95 tensor argument. In this case mark it with `category_override: factory`