Searched refs:_test_softmax (Results 1 – 2 of 2) sorted by relevance
22 def _test_softmax(self, inputs): member in TestSoftmax42 self._test_softmax(inputs)46 self._test_softmax(inputs)
1255 def _test_softmax(self, device): member in TestTensorExprFuser1290 self._test_softmax('cpu')1295 self._test_softmax('cuda')