Browse Source

support unknown control op subgraph

pull/1330/head
lichun 4 years ago
parent
commit
bc0cb4d9be
1 changed files with 13 additions and 0 deletions
  1. +13
    -0
      tests/ut/ge/common/format_transfer_transpose_unittest.cc

+ 13
- 0
tests/ut/ge/common/format_transfer_transpose_unittest.cc View File

@@ -4683,6 +4683,19 @@ TEST_F(UtestFormatTranspose, invalid_src_data) {
FormatTransferTranspose transpose;
TransResult result;
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 ge

Loading…
Cancel
Save