From 52f4face118180faeff833e842d7bf93a903c8b9 Mon Sep 17 00:00:00 2001
From: elade <elade@google.com>
Date: Thu, 13 Jun 2019 00:34:08 -0700
Subject: [PATCH] Remove int32 cast from StructureExporter.

PiperOrigin-RevId: 252976884
---
 morph_net/tools/structure_exporter.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/morph_net/tools/structure_exporter.py b/morph_net/tools/structure_exporter.py
index ca17479..17f17e7 100644
--- a/morph_net/tools/structure_exporter.py
+++ b/morph_net/tools/structure_exporter.py
@@ -70,8 +70,7 @@ def __init__(self,
         tf.logging.warning('No regularizer found for: %s', op.name)
         continue
 
-      self._tensors[rename_fn(op.name)] = tf.cast(op_regularizer.alive_vector,
-                                                  tf.int32)
+      self._tensors[rename_fn(op.name)] = op_regularizer.alive_vector
 
   @property
   def tensors(self):