Searched defs:gradcheck (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_sparse.py | 63 gradcheck = functools.partial(gradcheck, check_batched_grad=False) variable
459 def test_to_dense_with_gradcheck(self, device, dtype, gradcheck): argument
595 def test_to_dense_hybrid(self, device, dtype, gradcheck): argument
951 def test_permute(self, device, dtype, coalesced, gradcheck): argument
1621 def test_sparse_mul(self, device, dtype, coalesced, gradcheck): argument
4798 def test_gradcheck_to_dense(self, from_layout, device, dtype, index_dtype, gradcheck): argument
|
| H A D | test_mkldnn.py | 31 gradcheck = functools.partial(gradcheck, check_batched_grad=False) variable
|
| /aosp_15_r20/external/pytorch/torch/sparse/ |
| H A D | __init__.py | 549 def as_sparse_gradcheck(gradcheck): argument
|
| /aosp_15_r20/external/pytorch/test/distributed/ |
| H A D | test_data_parallel.py | 35 gradcheck = functools.partial(gradcheck, check_batched_grad=False) variable
|
| /aosp_15_r20/external/pytorch/torch/autograd/ |
| H A D | gradcheck.py | 1950 def gradcheck( function
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_utils.py | 4731 def gradcheck(fn, inputs, **kwargs): function
|