From f6ba21ed1d2cc8c5ad49a22c9b094f6f30c70c41 Mon Sep 17 00:00:00 2001 From: unknown Date: Thu, 4 Mar 2021 17:28:24 +0800 Subject: [PATCH] Add ut. --- .../graph/passes/transpose_transdata_pass_unittest.cc | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/tests/ut/ge/graph/passes/transpose_transdata_pass_unittest.cc b/tests/ut/ge/graph/passes/transpose_transdata_pass_unittest.cc index 7fc5973a..07919dc6 100644 --- a/tests/ut/ge/graph/passes/transpose_transdata_pass_unittest.cc +++ b/tests/ut/ge/graph/passes/transpose_transdata_pass_unittest.cc @@ -30,7 +30,7 @@ #include "graph/debug/ge_attr_define.h" namespace ge { -class UtestGraphPassesTransposeTransDataPass : public testing::Test { +class UtestGraphPassesTransposeTransdataPass : public testing::Test { protected: void SetUp() {} void TearDown() {} @@ -40,11 +40,11 @@ static ComputeGraphPtr BuildGraphTransposeD() { auto builder = ut::GraphBuilder("g1"); auto transdata1 = builder.AddNode("transdata1", "TransData", 1, 1, FORMAT_NC1HWC0, DT_FLOAT, std::vector({1, 1, 224, 224, 16})); transdata1->GetOpDesc()->MutableOutputDesc(0)->SetFormat(FORMAT_NHWC); - transdata1->GetOpDesc()->MutableOutputDesc(0)->SetShape(GeShape(std::vector({1, 1, 224, 224, 3}))); + transdata1->GetOpDesc()->MutableOutputDesc(0)->SetShape(GeShape(std::vector({1, 224, 224, 3}))); auto transpose1 = builder.AddNode("transpose1", "TransposeD", 1, 1, FORMAT_NCHW, DT_FLOAT, std::vector({1, 3, 224, 224})); - transpose1->GetOpDesc()->MutableIntputDesc(0)->SetFormat(FORMAT_NHWC); - transpose1->GetOpDesc()->MutableIntputDesc(0)->SetShape(GeShape(std::vector({1, 1, 224, 224, 3}))); + transpose1->GetOpDesc()->MutableInputDesc(0)->SetFormat(FORMAT_NHWC); + transpose1->GetOpDesc()->MutableInputDesc(0)->SetShape(GeShape(std::vector({1, 224, 224, 3}))); auto transdata2 = builder.AddNode("transdata2", "TransData", 1, 1, FORMAT_NCHW, DT_FLOAT, std::vector({1, 3, 224, 224})); transdata2->GetOpDesc()->MutableOutputDesc(0)->SetFormat(FORMAT_NC1HWC0); @@ -56,7 +56,7 @@ static ComputeGraphPtr BuildGraphTransposeD() { return builder.GetGraph(); } -TEST_F(UtestGraphPassesTransposeTransDataPass, test_run) { +TEST_F(UtestGraphPassesTransposeTransdataPass, test_run) { auto compute_graph = BuildGraphTransposeD(); compute_graph->SetSessionID(0);