add bigger shape test example for elementwise_pow OP (#22063)

release/1.7
chengjuntao 6 years ago committed by Tao Luo
parent 8b3ef2a587
commit e519a1fc82

@ -35,6 +35,26 @@ class TestElementwisePowOp(OpTest):
self.check_grad(['X', 'Y'], 'Out')
class TestElementwisePowOp_big_shape_1(TestElementwisePowOp):
def setUp(self):
self.op_type = "elementwise_pow"
self.inputs = {
'X': np.random.uniform(0.1, 1, [100, 100]).astype("float64"),
'Y': np.random.uniform(0.1, 1, [100, 100]).astype("float64")
}
self.outputs = {'Out': np.power(self.inputs['X'], self.inputs['Y'])}
class TestElementwisePowOp_big_shape_2(TestElementwisePowOp):
def setUp(self):
self.op_type = "elementwise_pow"
self.inputs = {
'X': np.random.uniform(0.1, 1, [100, 100]).astype("float64"),
'Y': np.random.uniform(0.1, 1, [100, 100]).astype("float64") * 20
}
self.outputs = {'Out': np.power(self.inputs['X'], self.inputs['Y'])}
class TestElementwisePowOp_scalar(TestElementwisePowOp):
def setUp(self):
self.op_type = "elementwise_pow"

Loading…
Cancel
Save