Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/functorch/
H A Dtest_ops.py37 from torch.testing._internal.autograd_function_db import autograd_function_db
408 @ops(op_db + additional_op_db + autograd_function_db, allowed_dtypes=(torch.float,))
547 @ops(op_db + additional_op_db + autograd_function_db, allowed_dtypes=(torch.float,))
736 @ops(op_db + additional_op_db + autograd_function_db, allowed_dtypes=(torch.float,))
841 @ops(op_db + additional_op_db + autograd_function_db, allowed_dtypes=(torch.float,))
1044 @ops(op_db + additional_op_db + autograd_function_db, allowed_dtypes=(torch.float,))
1185 @ops(op_db + additional_op_db + autograd_function_db, allowed_dtypes=(torch.float,))
1322 @ops(op_db + additional_op_db + autograd_function_db, allowed_dtypes=(torch.float,))
1380 @ops(op_db + additional_op_db + autograd_function_db, allowed_dtypes=(torch.float,))
1469 @ops(op_db + additional_op_db + autograd_function_db, allowed_dtypes=(torch.float,))
[all …]
H A Dcommon_utils.py17 from torch.testing._internal.autograd_function_db import autograd_function_db
462 all_opinfos = op_db + additional_op_db + autograd_function_db + custom_op_db
H A Dtest_vmap.py48 from torch.testing._internal.autograd_function_db import autograd_function_db
4329 op_db + additional_op_db + autograd_function_db + custom_op_db,
4393 op_db + additional_op_db + autograd_function_db + custom_op_db,
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dautograd_function_db.py505 autograd_function_db = [ variable
H A Dcustom_op_db.py12 from torch.testing._internal.autograd_function_db import (
/aosp_15_r20/external/pytorch/
H A D.lintrunner.toml1324 'torch/testing/_internal/autograd_function_db.py',