[dredd] apply suggestions

change rule shape to be enclosed in brackets change type to uint32_t seperate inference test list ONE-DCO-1.0-Signed-off-by: JuYoung Lee [email protected]
Samsung · Sep 25, 2024 · 51db89a · 51db89a
1 parent 98cd197
commit 51db89a
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/compiler/circle-inspect/src/Dump.cpp b/compiler/circle-inspect/src/Dump.cpp
@@ -260,16 +260,16 @@ void DumpTensorShape::run(std::ostream &os, const circle::Model *model,
     {
       const auto tensor = tensors->Get(i);
       auto shape = tensor->shape_signature() ? tensor->shape_signature() : tensor->shape();
-      os << reader.tensor_name(tensor) << " ";
-      for (int8_t i = 0; i < shape->size(); i++)
+      os << reader.tensor_name(tensor) << " [";
+      for (uint32_t i = 0; i < shape->size(); i++)
       {
         os << shape->Get(i);
         if (i != shape->size() - 1)
         {
           os << ",";
         }
       }
-      os << std::endl;
+      os << "]" << std::endl;
     }
   }
 }

diff --git a/compiler/circle2circle-dredd-recipe-test/test.lst b/compiler/circle2circle-dredd-recipe-test/test.lst
@@ -96,7 +96,6 @@ Add(PadV2_001 PASS substitute_padv2_to_pad)
 Add(Softmax_001 PASS decompose_softmax)
 Add(Softmax_002 PASS decompose_softmax)
 Add(StridedSlice_003 PASS substitute_strided_slice_to_reshape)
-Add(Inf_Pad_000 PASS)
 
 # CSE test
 
@@ -142,3 +141,6 @@ Add(REGRESS_ONNX_Conv_BN_Relu6_001 PASS
 
 Add(REGRESS_ONNX_Mul_Mul_000 PASS
       convert_nchw_to_nhwc)
+
+# SHAPE INFERENCE test
+Add(Inf_Pad_000 PASS)