lint

Signed-off-by: xadupre <[email protected]>

Author: xadupre

tests/run_pretrained_models.py

@@ -492,7 +492,7 @@ def run_tflite():
                     tf.profiler.experimental.start(self.tf_profile)
                 while time.time() < stop:
                     for _ in range(PERF_STEP):
-                        _ = concrete_func(**inputs)
+                        _ = self.concrete_function(**inputs)
                     n += PERF_STEP
                 if self.tf_profile is not None:
                     tf.profiler.experimental.stop()

tf2onnx/onnx_opset/nn.py

@@ -1793,6 +1793,7 @@ def version_11(cls, ctx, node, **kwargs):
             node.type = "Identity"
             ctx.replace_inputs(node, [data])
             return
+        cond = None
         if len(conditions) == 1:
             cond = conditions[0]
         if len(conditions) == 2:

tf2onnx/tf_utils.py

@@ -351,9 +351,9 @@ def read_tf_node_def_attrs(node_def, input_dtypes, input_shapes):
 # ignore the following attributes
 TF_IGNORED_NODE_ATTRS = {
     "T", "unknown_rank", "_class", "Tshape", "use_cudnn_on_gpu", "Index", "Tpaddings",
-    "TI", "Tparams", "Tindices", "Tlen", "Tdim", "Tin", "dynamic_size", "Tmultiples",
+    "TI", "Tparams", "Tindices", "Tlen", "Tdim", "dynamic_size", "Tmultiples",
     "Tblock_shape", "Tcrops", "index_type", "Taxis", "U", "maxval",
-    "Tout", "Tlabels", "Tindex", "element_shape", "Targmax", "Tperm", "Tcond",
+    "Tlabels", "Tindex", "element_shape", "Targmax", "Tperm", "Tcond",
     "T_threshold", "shape_type", "_lower_using_switch_merge",
     "parallel_iterations", "_num_original_outputs", "output_types", "output_shapes",
     "key_dtype", "value_dtype", "Tin", "Tout", "capacity", "component_types", "shapes",

tf2onnx/utils.py

@@ -46,7 +46,7 @@
     onnx_pb.TensorProto.UINT32: np.uint32,
     onnx_pb.TensorProto.UINT64: np.uint64,
     onnx_pb.TensorProto.INT64: np.int64,
-    onnx_pb.TensorProto.UINT64: np.uint64,
+    onnx_pb.TensorProto.INT64: np.int64,
     onnx_pb.TensorProto.BOOL: bool,
     onnx_pb.TensorProto.COMPLEX64: np.complex64,
     onnx_pb.TensorProto.COMPLEX128: np.complex128,