Searched refs:_test_arithmetic (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_legacy_vmap.py | 2620 def _test_arithmetic(self, op, device, test_grad_grad=True): member in TestVmapBatchedGradientLegacy
2632 self._test_arithmetic(torch.add, device, test_grad_grad=False)
2633 self._test_arithmetic(lambda x, y: x + y, device, test_grad_grad=False)
2636 self._test_arithmetic(torch.sub, device, test_grad_grad=False)
2637 self._test_arithmetic(lambda x, y: x - y, device, test_grad_grad=False)
2640 self._test_arithmetic(torch.mul, device)
2641 self._test_arithmetic(lambda x, y: x * y, device)
2644 self._test_arithmetic(torch.div, device)
2645 self._test_arithmetic(lambda x, y: x / y, device)
|
| /aosp_15_r20/external/pytorch/test/functorch/ |
| H A D | test_vmap.py | 3692 def _test_arithmetic(self, op, device, test_grad_grad=True): member in TestVmapBatchedGradient
3704 self._test_arithmetic(torch.add, device, test_grad_grad=False)
3705 self._test_arithmetic(lambda x, y: x + y, device, test_grad_grad=False)
3708 self._test_arithmetic(torch.sub, device, test_grad_grad=False)
3709 self._test_arithmetic(lambda x, y: x - y, device, test_grad_grad=False)
3712 self._test_arithmetic(torch.mul, device)
3713 self._test_arithmetic(lambda x, y: x * y, device)
3716 self._test_arithmetic(torch.div, device)
3717 self._test_arithmetic(lambda x, y: x / y, device)
|