Fix(//core/conversion/InterfaceTypes.cpp): Fix dimension padding bug, changed toDimsPad() to toDims()

abhi-iyer · abhi-iyer · commit b1d503507145 · 2020-06-08T11:13:54.000-07:00
Signed-off-by: Abhiram Iyer &lt;abhirami@nvidia.com&gt;
Signed-off-by: Abhiram Iyer &lt;abhi.iyer.ai@gmail.com&gt;
diff --git a/core/conversion/InterfaceTypes.cpp b/core/conversion/InterfaceTypes.cpp
@@ -55,9 +55,9 @@ InputRange::InputRange(std::vector<int64_t> min_shape, std::vector<int64_t> opt_
                   << max_shape.size() << ")");
     }
 
-    min = util::toDimsPad(min_shape, 4);
-    opt = util::toDimsPad(opt_shape, 4);
-    max = util::toDimsPad(max_shape, 4);
+    min = util::toDims(min_shape);
+    opt = util::toDims(opt_shape);
+    max = util::toDims(max_shape);
 
     std::vector<int64_t> dyn_shape;
     for (size_t i = 0; i < opt_shape.size(); i++) {
@@ -72,7 +72,7 @@ InputRange::InputRange(std::vector<int64_t> min_shape, std::vector<int64_t> opt_
         }
     }
 
-    input_shape = util::toDimsPad(dyn_shape, 4);
+    input_shape = util::toDims(dyn_shape);
 
 }
 

Original file line number	Diff line number	Diff line change
`@@ -55,9 +55,9 @@ InputRange::InputRange(std::vector<int64_t> min_shape, std::vector<int64_t> opt_`
`55`	`55`	`<< max_shape.size() << ")");`
`56`	`56`	`}`
`57`	`57`
`58`		`- min = util::toDimsPad(min_shape, 4);`
`59`		`- opt = util::toDimsPad(opt_shape, 4);`
`60`		`- max = util::toDimsPad(max_shape, 4);`
	`58`	`+ min = util::toDims(min_shape);`
	`59`	`+ opt = util::toDims(opt_shape);`
	`60`	`+ max = util::toDims(max_shape);`
`61`	`61`
`62`	`62`	`std::vector<int64_t> dyn_shape;`
`63`	`63`	`for (size_t i = 0; i < opt_shape.size(); i++) {`
`@@ -72,7 +72,7 @@ InputRange::InputRange(std::vector<int64_t> min_shape, std::vector<int64_t> opt_`
`72`	`72`	`}`
`73`	`73`	`}`
`74`	`74`
`75`		`- input_shape = util::toDimsPad(dyn_shape, 4);`
	`75`	`+ input_shape = util::toDims(dyn_shape);`
`76`	`76`
`77`	`77`	`}`
`78`	`78`