Skip to content

Commit 7b7f131

Browse files
committed
refactor(//tests): Update batchnorm tests for new stride IR
Signed-off-by: Naren Dasan <[email protected]> Signed-off-by: Naren Dasan <[email protected]>
1 parent d6a3c45 commit 7b7f131

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

tests/core/conversion/converters/test_batch_norm.cpp

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -38,10 +38,10 @@ TEST(Converters, ATenBatchNormConvertsCorrectly) {
3838
TEST(Converters, ATenBatchNormShouldUnpackConvertsCorrectly) {
3939
const auto graph = R"IR(
4040
graph(%0 : Tensor,
41-
%1: Float(5:1),
42-
%2: Float(5:1),
43-
%3: Float(5:1),
44-
%4: Float(5:1)):
41+
%1: Float(5, strides=[1]),
42+
%2: Float(5, strides=[1]),
43+
%3: Float(5, strides=[1]),
44+
%4: Float(5, strides=[1])):
4545
%5 : bool = prim::Constant[value=0]()
4646
%6 : float = prim::Constant[value=1.0000000000000001e-05]()
4747
%7 : float = prim::Constant[value=0.10000000000000001]()

0 commit comments

Comments
 (0)