- void getBatchSupportInfoImpl() const override {
- IE_ASSERT(_inputEdges.size() == 1);
- IE_ASSERT(_outputEdges.size() == 1);
-
- _batchInfo.setInput(_inputEdges[0], BatchSupport::Split);
- _batchInfo.setOutput(_outputEdges[0], BatchSupport::Split);
+ void getBatchSupportInfoImpl(StageDataInfo<BatchSupport>& batchInfo) override {
+ batchInfo.setInput(inputEdge(0), BatchSupport::Split);
+ batchInfo.setOutput(outputEdge(0), BatchSupport::Split);