diff options
Diffstat (limited to 'compiler/luci/pass/src/ReplaceNonConstFCWithBatchMatMulPass.test.cpp')
| -rw-r--r-- | compiler/luci/pass/src/ReplaceNonConstFCWithBatchMatMulPass.test.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/compiler/luci/pass/src/ReplaceNonConstFCWithBatchMatMulPass.test.cpp b/compiler/luci/pass/src/ReplaceNonConstFCWithBatchMatMulPass.test.cpp index 7606a6125..93024f3f7 100644 --- a/compiler/luci/pass/src/ReplaceNonConstFCWithBatchMatMulPass.test.cpp +++ b/compiler/luci/pass/src/ReplaceNonConstFCWithBatchMatMulPass.test.cpp @@ -159,8 +159,8 @@ TEST_F(ReplaceNonConstFCWithBatchMatMulPassTest, simple_test) auto ret = pass.run(g.g()); EXPECT_EQ(true, ret); - auto mm = dynamic_cast<luci::CircleBatchMatMul *>(g.output()->from()); - EXPECT_NE(nullptr, mm); + auto res = dynamic_cast<luci::CircleReshape *>(g.output()->from()); + EXPECT_NE(nullptr, res); } TEST_F(ReplaceNonConstFCWithBatchMatMulPassTest, nonzero_bias_test) |