diff options
author | 2017-11-21 23:55:59 -0800 | |
---|---|---|
committer | 2017-11-22 00:01:14 -0800 | |
commit | d0a3b2d3983b970b750329088013dc5cb67d96f9 (patch) | |
tree | 17cc584c4568e4e64a4bdd6bbee0be0b9d96f62c /tensorflow/python/kernel_tests/pooling_ops_test.py | |
parent | c6d603f02e1a98f871912cda6716cdcbed6b439e (diff) |
Merged commit includes the following changes:
176617057 by yifeif:
Internal change.
--
176615737 by yifeif:
Fix internal tests.
--
PiperOrigin-RevId: 176617057
Diffstat (limited to 'tensorflow/python/kernel_tests/pooling_ops_test.py')
-rw-r--r-- | tensorflow/python/kernel_tests/pooling_ops_test.py | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/tensorflow/python/kernel_tests/pooling_ops_test.py b/tensorflow/python/kernel_tests/pooling_ops_test.py index 150e2ff7f2..6be8997cab 100644 --- a/tensorflow/python/kernel_tests/pooling_ops_test.py +++ b/tensorflow/python/kernel_tests/pooling_ops_test.py @@ -18,8 +18,8 @@ from __future__ import absolute_import from __future__ import division from __future__ import print_function -import numpy as np import os +import numpy as np from tensorflow.python.framework import constant_op from tensorflow.python.framework import dtypes @@ -1442,7 +1442,6 @@ class PoolingTest(test.TestCase): use_gpu=True, v2=v2) - # Propagate the diff in cases of NaNs os.environ["TF_ENABLE_MAXPOOL_NANPROP"] = "1" expected_input_backprop_cudnn = expected_input_backprop_tf_cpu |