Skip to content

Commit 0e90f78

Browse files
committed
fix(//tests/core/converters/activations): Complete tensors in prelu test
Signed-off-by: Naren Dasan <[email protected]> Signed-off-by: Naren Dasan <[email protected]>
1 parent 02299b7 commit 0e90f78

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/core/converters/test_activation.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -112,7 +112,7 @@ TEST(Converters, ATenHardTanhCustomRangeConvertsCorrectly) {
112112
TEST(Converters, ATenPReLUConvertsCorrectly) {
113113
const auto graph = R"IR(
114114
graph(%0 : Tensor,
115-
%1 : Float(1)):
115+
%1 : Float(1:1)):
116116
%3 : Tensor = aten::prelu(%0, %1)
117117
return (%3))IR";
118118

@@ -135,7 +135,7 @@ TEST(Converters, ATenPReLUConvertsCorrectly) {
135135
TEST(Converters, ATenPReLUMultiChannelConvertsCorrectly) {
136136
const auto graph = R"IR(
137137
graph(%0 : Tensor,
138-
%1 : Float(10)):
138+
%1 : Float(10:1)):
139139
%3 : Tensor = aten::prelu(%0, %1)
140140
return (%3))IR";
141141

0 commit comments

Comments
 (0)