Skip to content

Commit b02c316

Browse files
pkchmn-robot
authored andcommitted
Internal change
PiperOrigin-RevId: 298946832
1 parent ec2f6b8 commit b02c316

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

morph_net/tools/configurable_ops.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -48,10 +48,10 @@
4848
import json
4949
from enum import Enum
5050

51-
import tensorflow as tf
51+
import tensorflow.compat.v1 as tf
5252
from tensorflow.contrib import framework
5353
from tensorflow.contrib import layers
54-
gfile = tf.gfile # Aliase needed for mock.
54+
# gfile = tf.gfile # Aliase needed for mock.
5555

5656
VANISHED = 0.0
5757
NUM_OUTPUTS = 'num_outputs'
@@ -514,7 +514,7 @@ def decorator_from_parameterization_file(
514514
Returns:
515515
An ConfigurableOps instance with the parameterization from `filename`.
516516
"""
517-
with gfile.Open(filename, 'r') as f:
517+
with tf.gfile.Open(filename, 'r') as f:
518518
parameterization = json.loads(f.read())
519519
return ConfigurableOps(
520520
parameterization=parameterization, fallback_rule=fallback_rule)

0 commit comments

Comments
 (0)