fix issue with randomop

Author: sdpython

benchmarks/profile_conversion_time.py

@@ -33,9 +33,9 @@ def spy_convert(graph_def, model):
         tf.import_graph_def(graph_def=graph_def, name='')
 
         def spy_convert_in():
-            return tfonnx.process_tf_graph(tf_graph=graph,
-                                    input_names=[model.input.name],
-                                    output_names=[model.output.name])
+            return tfonnx.process_tf_graph(
+                tf_graph=graph, input_names=[model.input.name],
+                output_names=[model.output.name])
 
         spy_convert_in()
 
@@ -55,19 +55,21 @@ def convert(graph_def, model):
     spy_convert(graph_def, model)
 
 
-def profile(profiler="pyinstrument", name="MobileNet", show_all=False,
+def profile(profiler="none", name="MobileNet", show_all=False,
             module='tf.keras'):
     """
     Profiles the conversion of a model.
-    
-    :param profiler: one among spy, pyinstrument, cProfile
+
+    :param profiler: one among none, spy, pyinstrument, cProfile
     :param name: model to profile, MobileNet, EfficientNetB2
     :param show_all: use by pyinstrument to show all functions
     """
     print("create(%r, %r, %r)" % (profiler, name, module))
     graph_def, model = create(name, module)
     print("profile(%r, %r, %r)" % (profiler, name, module))
-    if profiler == "spy":
+    if profiler == 'none':
+        convert(graph_def, model)
+    elif profiler == "spy":
         # py-spy record -r 10 -o profile.svg -- python conversion_time.py spy
         convert(graph_def, model)
     elif profiler == "pyinstrument":

tf2onnx/graph.py

@@ -282,7 +282,7 @@ def get_tensor_value(self, as_list=True):
                 when as_list=True, return 1, type is <class 'int'>.
         """
         if not self.is_const():
-            raise ValueError("get tensor value: {} must be Const".format(self.name))
+            raise ValueError("get tensor value: '{}' must be Const".format(self.name))
 
         t = self.get_attr("value")
         if t:

tf2onnx/onnx_opset/generator.py

@@ -38,7 +38,7 @@ def version_1(cls, ctx, node, **kwargs):
         # const we make it an attribute.
         seed = node.get_attr("seed")
         node.set_attr("seed", float(seed.f))
-        if len(node.input) > 0:
+        if len(node.input) > 0 and node.inputs[0].is_const():
             shape = node.inputs[0].get_tensor_value()
             ctx.remove_input(node, node.input[0], 0)
             node.set_attr("shape", shape)

tf2onnx/tfonnx.py

@@ -257,7 +257,8 @@ def tensorflow_onnx_mapping(g, ops_mapping):
             func(g, node, **kwargs)
             node.skip_conversion = True
         except Exception as ex:
-            logger.error("Failed to convert node %s\n%s", node.name, node.summary, exc_info=1)
+            logger.error("Failed to convert node %r (fct=%r)\n%r",
+                node.name, func, node.summary, exc_info=1)
             exceptions.append(ex)
 
     return mapped_op, unmapped_op, exceptions
@@ -450,14 +451,28 @@ def compat_handler(ctx, node, **kwargs):
 
     # pre-processing graph rewrites
     # bi-directional re-writer should be placed after single directional re-writer
-    rewriters = [rewrite_constant_fold, rewrite_quantize_and_dequantize, rewrite_transpose, rewrite_flatten,
-                 rewrite_random_uniform, rewrite_random_uniform_fold_const,
-                 rewrite_random_normal, rewrite_dropout, rewrite_eye,
-                 rewrite_leakyrelu, rewrite_thresholded_relu, rewrite_conv2d_with_pad,
-                 rewrite_single_direction_lstm, rewrite_bi_direction_lstm,
-                 rewrite_single_direction_gru, rewrite_bi_direction_gru,
-                 rewrite_custom_rnn_cell, rewrite_generic_loop, rewrite_cond,
-                 rewrite_biasadd_with_conv2d, rewrite_gemm
+    rewriters = [# single directional
+                 rewrite_constant_fold,
+                 rewrite_quantize_and_dequantize,
+                 rewrite_transpose,
+                 rewrite_flatten,
+                 rewrite_random_uniform,
+                 rewrite_random_uniform_fold_const,
+                 rewrite_random_normal,
+                 rewrite_dropout,
+                 rewrite_eye,
+                 rewrite_leakyrelu,
+                 rewrite_thresholded_relu,
+                 rewrite_conv2d_with_pad,
+                 rewrite_single_direction_lstm,
+                 # bi-directional
+                 rewrite_bi_direction_lstm,
+                 rewrite_single_direction_gru,
+                 rewrite_bi_direction_gru,
+                 rewrite_custom_rnn_cell,
+                 rewrite_generic_loop, rewrite_cond,
+                 rewrite_biasadd_with_conv2d,
+                 rewrite_gemm,
                  ]
 
     if custom_rewriter is not None: