diff --git a/mindspore/ccsrc/dataset/engine/datasetops/dataset_op.h b/mindspore/ccsrc/dataset/engine/datasetops/dataset_op.h index 21b5cbf7ce3bf19b014d04ec59dd8efac6bcc679..b47ceea6904b10959a16a70ca2180771775a2d8f 100644 --- a/mindspore/ccsrc/dataset/engine/datasetops/dataset_op.h +++ b/mindspore/ccsrc/dataset/engine/datasetops/dataset_op.h @@ -217,8 +217,8 @@ class DatasetOp : public std::enable_shared_from_this { if (!inlined()) { return out_connector_->size(); } - // Return -1 for inlined op - return -1; + // Return child connector size for inlined op + return ChildOpConnectorSize(); } // Getter function @@ -227,8 +227,8 @@ class DatasetOp : public std::enable_shared_from_this { if (!inlined()) { return out_connector_->capacity(); } - // Return -1 for inlined op - return -1; + // Return child connector capacity for inlined op + return ChildOpConnectorCapacity(); } // Getter function