_nll_loss_helper (reference) in projects: aosp_15_r20

Project(s)

Full Search
Definition
Symbol
File Path
History
Type

Searched refs:_nll_loss_helper (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/test/
H A D	test_mps.py	8607 def _nll_loss_helper(self, input_size, reduction, expected): member in TestNLLLoss 8654 self._nll_loss_helper([1, 3], "none", torch.empty([0], device=device)) 8655 self._nll_loss_helper([3, 5, 7], "none", torch.empty([5, 7], device=device)) 8656 self._nll_loss_helper([2, 3, 1, 7], "none", torch.empty([2, 1, 7], device=device)) 8657 self._nll_loss_helper([2, 3, 5, 1], "none", torch.empty([2, 5, 1], device=device)) 8658 self._nll_loss_helper([2, 3, 5, 7, 1], "none", torch.empty([2, 5, 7, 1], device=device)) 8662 self._nll_loss_helper([1, 3], "mean", nan) 8663 self._nll_loss_helper([1, 3, 5, 7], "mean", nan) 8664 self._nll_loss_helper([2, 3, 1, 7], "mean", nan) 8665 self._nll_loss_helper([2, 3, 5, 1], "mean", nan) [all …]
H A D	test_nn.py	11531 def _nll_loss_helper(self, input_size, reduction, expected, device): member in TestNNDeviceType 11544 self._nll_loss_helper([0, 3], "none", torch.empty([0], device=device), device) 11545 self._nll_loss_helper([0, 3, 5, 7], "none", torch.empty([0, 5, 7], device=device), device) 11546 self._nll_loss_helper([2, 3, 0, 7], "none", torch.empty([2, 0, 7], device=device), device) 11547 self._nll_loss_helper([2, 3, 5, 0], "none", torch.empty([2, 5, 0], device=device), device) 11548 … self._nll_loss_helper([2, 3, 5, 7, 0], "none", torch.empty([2, 5, 7, 0], device=device), device) 11553 self._nll_loss_helper([0, 3], "mean", nan, device) 11554 self._nll_loss_helper([0, 3, 5, 7], "mean", nan, device) 11555 self._nll_loss_helper([2, 3, 0, 7], "mean", nan, device) 11556 self._nll_loss_helper([2, 3, 5, 0], "mean", nan, device) [all …]