| @@ -4683,6 +4683,19 @@ TEST_F(UtestFormatTranspose, invalid_src_data) { | |||||
| FormatTransferTranspose transpose; | FormatTransferTranspose transpose; | ||||
| TransResult result; | TransResult result; | ||||
| EXPECT_EQ(transpose.TransFormat(args, result), ACL_ERROR_GE_PARAM_INVALID); | EXPECT_EQ(transpose.TransFormat(args, result), ACL_ERROR_GE_PARAM_INVALID); | ||||
| uint8_t data1[3] = {14583, 12849, 14184}; | |||||
| TransArgs args1{data1, FORMAT_NCHW, FORMAT_NHWC, std::vector<int64_t>({-1, 3, 1, 1}), std::vector<int64_t>({1, 1, 1, 3}), DT_INT64}; | |||||
| FormatTransferTranspose transpose1; | |||||
| TransResult result1; | |||||
| EXPECT_EQ(transpose1.TransFormat(args1, result1), ACL_ERROR_GE_SHAPE_INVALID); | |||||
| TransArgs args2{data1, FORMAT_NCHW, FORMAT_NHWC, std::vector<int64_t>({1, 3, 1, 1}), std::vector<int64_t>({1, 1, 1, 3}), DT_INT64}; | |||||
| FormatTransferTranspose transpose1; | |||||
| TransResult result1; | |||||
| EXPECT_EQ(transpose1.TransFormat(args1, result1), ACL_ERROR_GE_SHAPE_INVALID); | |||||
| } | } | ||||
| } // namespace formats | } // namespace formats | ||||
| } // namespace ge | } // namespace ge | ||||