Home
last modified time | relevance | path

Searched refs:is_factory_function (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/pytorch/tools/autograd/
H A Dgen_variable_factories.py46 factory_functions = [fn for fn in native_functions if is_factory_function(fn)]
63 def is_factory_function(f: NativeFunction) -> bool: function
/aosp_15_r20/external/pytorch/torchgen/api/
H A Dpython.py800 is_factory_function = category_override == "factory" or (
811 if (is_factory_function or is_like_or_new_function) and not is_dummy_function:
/aosp_15_r20/external/pytorch/test/
H A Dtest_ops.py1001 if op.is_factory_function and sample.kwargs.get("device", None) is None:
1042 if op.is_factory_function and sample.kwargs.get("dtype", None) is None:
1052 if op.supports_out and (op.supports_autograd or op.is_factory_function)
2646 @ops([op for op in ops_and_refs if op.is_factory_function])
H A Dtest_prims.py449 if op.is_factory_function:
/aosp_15_r20/external/pytorch/torch/testing/_internal/opinfo/
H A Dcore.py897 is_factory_function: bool = False variable in OpInfo
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dcommon_methods_invocations.py11760 is_factory_function=True,
11946 is_factory_function=False,
13609 is_factory_function=True,
13637 is_factory_function=True,
13665 is_factory_function=True,
13698 is_factory_function=True,
18755 is_factory_function=True,
18775 is_factory_function=True,
18795 is_factory_function=True,