Skip to content

Commit 85f3ca7

Browse files
pkchmn-robot
authored andcommitted
Remove unneeded cast to int32.
PiperOrigin-RevId: 252956325
1 parent 2a52143 commit 85f3ca7

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

morph_net/tools/structure_exporter.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -70,7 +70,8 @@ def __init__(self,
7070
tf.logging.warning('No regularizer found for: %s', op.name)
7171
continue
7272

73-
self._tensors[rename_fn(op.name)] = op_regularizer.alive_vector
73+
self._tensors[rename_fn(op.name)] = tf.cast(op_regularizer.alive_vector,
74+
tf.int32)
7475

7576
@property
7677
def tensors(self):

0 commit comments

Comments
 (0)