Remove int32 cast from StructureExporter.

PiperOrigin-RevId: 252976884

Author: eladeban

Diff for: 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):