diff options
| author |  Skye Wanderman-Milne <skyewm@google.com> | 2018-05-16 18:03:01 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-05-16 18:06:14 -0700 |
| commit | 0c33e1ff58e121a0cf6acdb6aeb8de53cb3abf25 (patch) | |
| tree | 42d8a74609e78bffda238c3f8f0538bfdf61a80c /tensorflow/python/grappler | |
| parent | 1511cf2f51965151147d1ca2e822439c6656b821 (diff) | |
Remove _USE_C_API staging in tests now that the C API is enabled by default.
This is in preparation for removing the _USE_C_API toggle altogether.
PiperOrigin-RevId: 196920481
Diffstat (limited to 'tensorflow/python/grappler')
| -rw-r--r-- | tensorflow/python/grappler/memory_optimizer_test.py | 3 |
1 files changed, 0 insertions, 3 deletions
diff --git a/tensorflow/python/grappler/memory_optimizer_test.py b/tensorflow/python/grappler/memory_optimizer_test.py index 3f9d8864a2..7ed4b128e4 100644 --- a/tensorflow/python/grappler/memory_optimizer_test.py +++ b/tensorflow/python/grappler/memory_optimizer_test.py @@ -25,7 +25,6 @@ from tensorflow.python.client import session from tensorflow.python.framework import meta_graph from tensorflow.python.framework import ops from tensorflow.python.framework import random_seed -from tensorflow.python.framework import test_util from tensorflow.python.grappler import tf_optimizer from tensorflow.python.ops import math_ops from tensorflow.python.ops import nn @@ -35,7 +34,6 @@ from tensorflow.python.platform import test from tensorflow.python.training import training as train -@test_util.with_c_api class MemoryOptimizerSwapTest(test.TestCase): """Tests the Grappler memory optimizer.""" @@ -96,7 +94,6 @@ class MemoryOptimizerSwapTest(test.TestCase): self.assertEqual('c', node.input[1]) -@test_util.with_c_api class MemoryOptimizerRecomputeTest(test.TestCase): """Tests the Python interface to recomputation rewrites. |