[tfdbg] Fix examples_test for TF2
RELNOTES: N/A PiperOrigin-RevId: 264270116
This commit is contained in:
parent
764857224b
commit
9ec481cd6d
@ -22,10 +22,12 @@ import sys
|
|||||||
import tempfile
|
import tempfile
|
||||||
|
|
||||||
import numpy as np
|
import numpy as np
|
||||||
import tensorflow as tf
|
import tensorflow
|
||||||
|
|
||||||
from tensorflow.python import debug as tf_debug
|
from tensorflow.python import debug as tf_debug
|
||||||
|
|
||||||
|
tf = tensorflow.compat.v1
|
||||||
|
|
||||||
|
|
||||||
def main(_):
|
def main(_):
|
||||||
sess = tf.Session()
|
sess = tf.Session()
|
||||||
|
@ -22,10 +22,12 @@ import sys
|
|||||||
|
|
||||||
import numpy as np
|
import numpy as np
|
||||||
from six.moves import xrange # pylint: disable=redefined-builtin
|
from six.moves import xrange # pylint: disable=redefined-builtin
|
||||||
import tensorflow as tf
|
import tensorflow
|
||||||
|
|
||||||
from tensorflow.python import debug as tf_debug
|
from tensorflow.python import debug as tf_debug
|
||||||
|
|
||||||
|
tf = tensorflow.compat.v1
|
||||||
|
|
||||||
FLAGS = None
|
FLAGS = None
|
||||||
|
|
||||||
|
|
||||||
|
@ -23,10 +23,12 @@ import sys
|
|||||||
import tempfile
|
import tempfile
|
||||||
|
|
||||||
import numpy as np
|
import numpy as np
|
||||||
import tensorflow as tf
|
import tensorflow
|
||||||
|
|
||||||
from tensorflow.python import debug as tf_debug
|
from tensorflow.python import debug as tf_debug
|
||||||
|
|
||||||
|
tf = tensorflow.compat.v1
|
||||||
|
|
||||||
|
|
||||||
def main(_):
|
def main(_):
|
||||||
# Create a dummy dataset.
|
# Create a dummy dataset.
|
||||||
|
@ -28,11 +28,13 @@ import argparse
|
|||||||
import sys
|
import sys
|
||||||
import tempfile
|
import tempfile
|
||||||
|
|
||||||
import tensorflow as tf
|
import tensorflow
|
||||||
|
|
||||||
from tensorflow.examples.tutorials.mnist import input_data
|
from tensorflow.examples.tutorials.mnist import input_data
|
||||||
from tensorflow.python import debug as tf_debug
|
from tensorflow.python import debug as tf_debug
|
||||||
|
|
||||||
|
tf = tensorflow.compat.v1
|
||||||
|
|
||||||
|
|
||||||
IMAGE_SIZE = 28
|
IMAGE_SIZE = 28
|
||||||
HIDDEN_SIZE = 500
|
HIDDEN_SIZE = 500
|
||||||
|
@ -21,10 +21,11 @@ import argparse
|
|||||||
import sys
|
import sys
|
||||||
import tempfile
|
import tempfile
|
||||||
|
|
||||||
import tensorflow as tf
|
import tensorflow
|
||||||
|
|
||||||
from tensorflow.python import debug as tf_debug
|
from tensorflow.python import debug as tf_debug
|
||||||
|
|
||||||
|
tf = tensorflow.compat.v1
|
||||||
|
|
||||||
_IRIS_INPUT_DIM = 4
|
_IRIS_INPUT_DIM = 4
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user