Searched refs:_check_zero_nnz_softmax_op (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_sparse.py | 3566 def _check_zero_nnz_softmax_op(self, func, ndim, device, dtype): member in TestSparse
3580 self._check_zero_nnz_softmax_op(torch.sparse.softmax, 1, device, dtype)
3581 self._check_zero_nnz_softmax_op(torch.sparse.softmax, 10, device, dtype)
3586 self._check_zero_nnz_softmax_op(torch.sparse.log_softmax, 1, device, dtype)
3587 self._check_zero_nnz_softmax_op(torch.sparse.log_softmax, 10, device, dtype)
|