fix(//tests/core/converters/activations): Complete tensors in prelu test

Signed-off-by: Naren Dasan <naren@narendasan.com> Signed-off-by: Naren Dasan <narens@nvidia.com>
pytorch · Aug 4, 2020 · 0e90f78 · 0e90f78
1 parent 02299b7
commit 0e90f78
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/core/converters/test_activation.cpp b/tests/core/converters/test_activation.cpp
@@ -112,7 +112,7 @@ TEST(Converters, ATenHardTanhCustomRangeConvertsCorrectly) {
 TEST(Converters, ATenPReLUConvertsCorrectly) {
     const auto graph = R"IR(
       graph(%0 : Tensor,
-            %1 : Float(1)):
+            %1 : Float(1:1)):
         %3 : Tensor = aten::prelu(%0, %1)
         return (%3))IR";
 
@@ -135,7 +135,7 @@ TEST(Converters, ATenPReLUConvertsCorrectly) {
 TEST(Converters, ATenPReLUMultiChannelConvertsCorrectly) {
     const auto graph = R"IR(
       graph(%0 : Tensor,
-            %1 : Float(10)):
+            %1 : Float(10:1)):
         %3 : Tensor = aten::prelu(%0, %1)
         return (%3))IR";