Searched refs:_spawn_test_multinomial_invalid_probs_cuda (Results 1 – 1 of 1) sorted by relevance
899 def _spawn_test_multinomial_invalid_probs_cuda(self, probs): member in TestCuda945 self._spawn_test_multinomial_invalid_probs_cuda([1.0, -1.0, 1.0])946 self._spawn_test_multinomial_invalid_probs_cuda([1.0, inf, 1.0])947 self._spawn_test_multinomial_invalid_probs_cuda([1.0, -inf, 1.0])948 self._spawn_test_multinomial_invalid_probs_cuda([1.0, 1.0, nan])