Fix 2 bugs to avoid extranious warnings (#1292)

Signed-off-by: Tom Wildenhain <[email protected]>

Author: TomWildenhain-Microsoft

tf2onnx/onnx_opset/nn.py

@@ -816,7 +816,7 @@ def version_6(cls, ctx, node, **kwargs):
 
             pop_var_squeezed = ctx.make_node("Div", [var_squeezed, cnt_float.output[0]]).output[0]
             ctx.replace_inputs(node, node.input[:3] + [avg_squeezed, pop_var_squeezed])
-        else:
+        elif is_training:
             logger.warning("Node %s of type %s has is_training set to true, which is not supperted. "
                            "Please re-save the model with training set to false.",
                            node.name, tf_type)

tf2onnx/onnx_opset/tensor.py

@@ -1194,10 +1194,11 @@ def version_1(cls, ctx, node, **kwargs):
         # for each output we need to squeeze axis
         for n in node.output:
             op_name = utils.make_name(node.name)
-            squeeze_node = GraphBuilder(ctx).make_squeeze({'data': n, 'axes': [axis]}, name=op_name, return_node=True)
+            shape = ctx.get_shape(n)
+            dtype = ctx.get_dtype(n)
+            squeeze_node = GraphBuilder(ctx).make_squeeze(
+                {'data': n, 'axes': [axis]}, name=op_name, return_node=True, shapes=[shape], dtypes=[dtype])
             ctx.insert_node_on_output(squeeze_node, n)
-            ctx.copy_shape(n, squeeze_node.output[0])
-            ctx.copy_dtype(n, squeeze_node.output[0])
 
         # split node is 1 rank higher than squeeze nodes
         output_shape = ctx.get_shape(node.output[0])