Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

Commit

Permalink
Delete non_rectified_operation_test
Browse files Browse the repository at this point in the history
  • Loading branch information
luobao-intel committed Jul 18, 2018
1 parent 78fc94c commit 56dca95
Showing 1 changed file with 1 addition and 20 deletions.
21 changes: 1 addition & 20 deletions tests/python/mkl/test_mkldnn.py
Original file line number Diff line number Diff line change
Expand Up @@ -92,7 +92,7 @@ def __getitem__(self, key):
# below line triggers different execution thread
for _ in loader:
y = net(mx.nd.array(np.ones(X))).asnumpy()
# output should be 016711406 (non-mkldnn mode output)
# output should be 016711406 (non-mkldnn mode output)
assert_almost_equal(y[0, 0, 0, 0], 0.016711406)
break

Expand Down Expand Up @@ -256,25 +256,6 @@ def check_softmax_training(stype):
for stype in stypes:
check_softmax_training(stype)

@with_seed()
def test_SoftmaxOutput():
def check_SoftmaxOutput_training(stype):
for shape in [(10, 3)]:
data_tmp = np.random.normal(-0.1, 0.1, size=shape)
label_tmp = np.random.randint(2,size=shape[0])

data = mx.symbol.Variable('data', stype=stype)
label = mx.symbol.Variable('label',stype=stype)

in_location = [mx.nd.array(data_tmp).tostype(stype),mx.nd.array(label_tmp).tostype(stype)]

test = mx.symbol.SoftmaxOutput(data,label)
check_numeric_gradient(test, in_location, numeric_eps=1e-2,rtol=0.16, atol=1e-4)

stypes = ['row_sparse', 'default']
for stype in stypes:
check_SoftmaxOutput_training(stype)

@with_seed()
def test_pooling():
def check_pooling_training(stype):
Expand Down

0 comments on commit 56dca95

Please sign in to comment.