diff --git a/compiler/circle-inspect/src/Dump.cpp b/compiler/circle-inspect/src/Dump.cpp index 5bd18ea31d5..08238736be9 100644 --- a/compiler/circle-inspect/src/Dump.cpp +++ b/compiler/circle-inspect/src/Dump.cpp @@ -260,8 +260,8 @@ 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) @@ -269,7 +269,7 @@ void DumpTensorShape::run(std::ostream &os, const circle::Model *model, 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 index b1a3c727e6f..0f30bb59c98 100644 --- 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) diff --git a/res/TensorFlowLiteRecipes/Inf_Pad_000/test.rule b/res/TensorFlowLiteRecipes/Inf_Pad_000/test.rule index 62625045543..71c1536f2e1 100644 --- a/res/TensorFlowLiteRecipes/Inf_Pad_000/test.rule +++ b/res/TensorFlowLiteRecipes/Inf_Pad_000/test.rule @@ -1,6 +1,6 @@ -# To check if dynamic dimension properly infered +# To check if dynamic dimension properly inferred RULE "VERIFY_FILE_FORMAT" $(verify_file_format) '=' 1 RULE "PAD_EXIST" $(op_count PAD) '=' 1 -RULE "PAD_SHAPE" $(tensor_shape ofm) '=' 1,-1,7,2 +RULE "PAD_SHAPE" $(tensor_shape ofm) '=' [1,-1,7,2]