fix the scalar shape check (#1223)

intel · Sep 8, 2022 · d625648 · d625648
1 parent ddeb9f5
commit d625648
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/neural_compressor/adaptor/tf_utils/graph_converter.py b/neural_compressor/adaptor/tf_utils/graph_converter.py
@@ -197,6 +197,9 @@ def _inference(self, model):
                     # sometimes the input_tensor is not the same order with inputs
                     # we should check and pair them
                     def check_shape(tensor, data):
+                        # scalar default True
+                        if tensor.shape == None:
+                            return True
                         tensor_shape = tuple(tensor.shape)
                         data_shape = tuple(data.shape)
                         for tensor_dim, data_dim in zip(tensor_shape, data_shape):