diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-06-28 17:11:05 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-06-28 17:13:56 -0700 |
| commit | 6dc9977e1dffc9558835cf4d0a62b61b6c85cc19 (patch) | |
| tree | c1e8bcbb2342df806d1387982370ce0b4a03c1ab /tensorflow/compiler/tests/conv3d_test.py | |
| parent | c81830af5d488de600a4f62392c63059e310c017 (diff) | |
Import package xla_test instead of class XLATestCase.
PiperOrigin-RevId: 202572322
Diffstat (limited to 'tensorflow/compiler/tests/conv3d_test.py')
| -rw-r--r-- | tensorflow/compiler/tests/conv3d_test.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/tensorflow/compiler/tests/conv3d_test.py b/tensorflow/compiler/tests/conv3d_test.py index 3bebf46511..31ee41f04f 100644 --- a/tensorflow/compiler/tests/conv3d_test.py +++ b/tensorflow/compiler/tests/conv3d_test.py @@ -21,7 +21,7 @@ from __future__ import print_function import numpy as np from six.moves import xrange # pylint: disable=redefined-builtin -from tensorflow.compiler.tests.xla_test import XLATestCase +from tensorflow.compiler.tests import xla_test from tensorflow.python.framework import constant_op from tensorflow.python.framework import dtypes from tensorflow.python.ops import array_ops @@ -33,7 +33,7 @@ from tensorflow.python.platform import googletest # Test cloned from # tensorflow/python/kernel_tests/conv3d_backprop_filter_v2_grad_test.py -class Conv3DBackpropFilterV2GradTest(XLATestCase): +class Conv3DBackpropFilterV2GradTest(xla_test.XLATestCase): def testGradient(self): with self.test_session(), self.test_scope(): @@ -66,7 +66,7 @@ class Conv3DBackpropFilterV2GradTest(XLATestCase): # Test cloned from tensorflow/python/kernel_tests/conv3d_transpose_test.py -class Conv3DTransposeTest(XLATestCase): +class Conv3DTransposeTest(xla_test.XLATestCase): def testConv3DTransposeSingleStride(self): with self.test_session(), self.test_scope(): |