diff options
Diffstat (limited to 'tensorflow/python/platform/test.py')
-rw-r--r-- | tensorflow/python/platform/test.py | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/tensorflow/python/platform/test.py b/tensorflow/python/platform/test.py new file mode 100644 index 0000000000..7d46f9cbc2 --- /dev/null +++ b/tensorflow/python/platform/test.py @@ -0,0 +1,6 @@ +from tensorflow.python.platform.googletest import GetTempDir +from tensorflow.python.platform.googletest import main +from tensorflow.python.framework.test_util import TensorFlowTestCase as TestCase +from tensorflow.python.framework.test_util import IsGoogleCudaEnabled as IsBuiltWithCuda + +get_temp_dir = GetTempDir |