chore: add in_shapes call

peri044 · peri044 · commit 38dc7d51cb54 · 2022-10-19T14:36:26.000-07:00
Signed-off-by: Dheeraj Peri &lt;peri.dheeraj@gmail.com&gt;
diff --git a/core/partitioning/segmentedblock/SegmentedBlock.h b/core/partitioning/segmentedblock/SegmentedBlock.h
@@ -82,9 +82,9 @@ struct SegmentedBlock {
       max_shapes_ = in_shapes;
     }
   }
-  // const std::vector<ir::Input>& in_shapes() const {
-  //   return in_shapes_;
-  // }
+  const std::vector<ir::Input>& in_shapes() const {
+    return opt_shapes_;
+  }
   void register_intypes(std::vector<at::ScalarType>& in_types) {
     in_types_ = in_types;
   }

Original file line number	Diff line number	Diff line change
`@@ -82,9 +82,9 @@ struct SegmentedBlock {`
`82`	`82`	`max_shapes_ = in_shapes;`
`83`	`83`	`}`
`84`	`84`	`}`
`85`		`- // const std::vector<ir::Input>& in_shapes() const {`
`86`		`- // return in_shapes_;`
`87`		`- // }`
	`85`	`+ const std::vector<ir::Input>& in_shapes() const {`
	`86`	`+ return opt_shapes_;`
	`87`	`+ }`
`88`	`88`	`void register_intypes(std::vector<at::ScalarType>& in_types) {`
`89`	`89`	`in_types_ = in_types;`
`90`	`90`	`}`