Searched refs:is_factory_function (Results 1 – 6 of 6) sorted by relevance
/aosp_15_r20/external/pytorch/tools/autograd/ |
H A D | gen_variable_factories.py | 46 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 D | python.py | 800 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 D | test_ops.py | 1001 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 D | test_prims.py | 449 if op.is_factory_function:
|
/aosp_15_r20/external/pytorch/torch/testing/_internal/opinfo/ |
H A D | core.py | 897 is_factory_function: bool = False variable in OpInfo
|
/aosp_15_r20/external/pytorch/torch/testing/_internal/ |
H A D | common_methods_invocations.py | 11760 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,
|