blob: bb07578cbe24e1f2976184937ba1ca5815691f12 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
|
"""Switch between depending on googletest or unittest."""
from __future__ import absolute_import
from __future__ import division
from __future__ import print_function
# pylint: disable=unused-import
# pylint: disable=g-import-not-at-top
# pylint: disable=wildcard-import
import tensorflow.python.platform
from . import control_imports
if control_imports.USE_OSS and control_imports.OSS_GOOGLETEST:
from tensorflow.python.platform.default._googletest import *
else:
from tensorflow.python.platform.google._googletest import *
|
