|
|
|
@@ -101,10 +101,8 @@ def test_pow(): |
|
|
|
result = testpow(input_tensor, power) |
|
|
|
assert np.all(result.asnumpy() == expect) |
|
|
|
net = PowNet() |
|
|
|
with pytest.raises(TypeError): |
|
|
|
net(input_tensor, True) |
|
|
|
with pytest.raises(TypeError): |
|
|
|
net(input_tensor, power2) |
|
|
|
net(input_tensor, True) |
|
|
|
net(input_tensor, power2) |
|
|
|
|
|
|
|
|
|
|
|
def test_exp(): |
|
|
|
|