experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Internal change.

Browse Source 
Change: 155301612
This commit is contained in:
Anna R 2017-05-06 18:07:31 -08:00 committed by TensorFlower Gardener
parent 7cac7f24d1
commit 93572de9a1
1 changed files with 5 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
tensorflow/python/kernel_tests/bias_op_test.py
View File

@ -149,10 +149,8 @@ class BiasAddTest(test.TestCase):
# Test gradient of BiasAddGrad
bias_add_grad = gradients_impl.gradients(
nn_ops.l2_loss(output_tensor), bias_tensor)[0]
# pylint: disable=unused-variable
grad_jacob_t, grad_jacob_n = gradient_checker.compute_gradient(
output_tensor, np_input.shape, bias_add_grad, bias.shape)
# pylint: enable=unused-variable
if dtype == np.float16:
# Compare fp16 theoretical gradients to fp32 numerical gradients,
@ -186,10 +184,11 @@ class BiasAddTest(test.TestCase):
if dtype == dtypes.float64:
threshold = 1e-10
self.assertAllClose(tensor_jacob_t, tensor_jacob_n, threshold, threshold)
# TODO(annarev): Re-add assertion for float16, float32 dtypes and NCHW
# once we figure out why this check started failing with cuda mavx.
if dtype == dtypes.float64 or data_format != "NCHW":
self.assertAllClose(bias_jacob_t, bias_jacob_n, threshold, threshold)
# TODO(annarev): Re-add assertion for grad_jacob_t and grad_jacob_n once
# we figure out why this check started failing with cuda mavx.
# self.assertAllClose(grad_jacob_t, grad_jacob_n, threshold, threshold)
self.assertAllClose(grad_jacob_t, grad_jacob_n, threshold, threshold)
def testGradientTensor(self):
for (data_format, use_gpu) in GetTestConfigs():