Skip to content

Commit 4958d2d

Browse files
BrianWiedertensorflower-gardener
authored andcommitted
Redirect usages of convert_variables_to_constants from graph_util_impl.py to convert_to_constants.py to remove a cycle.
PiperOrigin-RevId: 507831958
1 parent b211206 commit 4958d2d

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

tensorflow/examples/speech_commands/freeze.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,6 @@
4242

4343
import input_data
4444
import models
45-
from tensorflow.python.framework import graph_util
4645
from tensorflow.python.ops import gen_audio_ops as audio_ops
4746

4847
# If it's available, load the specialized feature generator. If this doesn't
@@ -227,7 +226,7 @@ def main(_):
227226
models.load_variables_from_checkpoint(sess, FLAGS.start_checkpoint)
228227

229228
# Turn all the variables into inline constants inside the graph and save it.
230-
frozen_graph_def = graph_util.convert_variables_to_constants(
229+
frozen_graph_def = tf.compat.v1.graph_util.convert_variables_to_constants(
231230
sess, sess.graph_def, ['labels_softmax'])
232231

233232
if FLAGS.save_format == 'graph_def':

0 commit comments

Comments
 (0)