blob: 7d46f9cbc2a67574f9d7e78ab0bd9973fff32b75 (
plain)
1
2
3
4
5
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
|