Searched refs:autograd_function_db (Results 1 – 6 of 6) sorted by relevance
/aosp_15_r20/external/pytorch/test/functorch/ |
H A D | test_ops.py | 37 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 D | common_utils.py | 17 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 D | test_vmap.py | 48 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 D | autograd_function_db.py | 505 autograd_function_db = [ variable
|
H A D | custom_op_db.py | 12 from torch.testing._internal.autograd_function_db import (
|
/aosp_15_r20/external/pytorch/ |
H A D | .lintrunner.toml | 1324 'torch/testing/_internal/autograd_function_db.py',
|