Example #1
0
 def check_backward(self, x_data, y_grad, axis=None, keepdims=False):
     gradient_check.check_backward(functions.Max(axis, keepdims),
                                   x_data,
                                   y_grad,
                                   eps=1e-5,
                                   rtol=1e-3,
                                   atol=1e-3)
Example #2
0
 def test_duplicate_axis(self):
     with self.assertRaises(ValueError):
         functions.Max((0, 0))
Example #3
0
 def test_invalid_axis_type_in_tuple(self):
     with self.assertRaises(TypeError):
         functions.Max((1, 'x'))
Example #4
0
 def test_invalid_axis_type(self):
     with self.assertRaises(TypeError):
         functions.Max([0])