fix profiling capacity

5 years ago · 53cee6589e
parent f1a004cfa2
commit 53cee6589e
1 changed files with 1 additions and 1 deletions
--- a/mindspore/ccsrc/dataset/engine/datasetops/dataset_op.h
+++ b/mindspore/ccsrc/dataset/engine/datasetops/dataset_op.h
@ -225,7 +225,7 @@ class DatasetOp : public std::enable_shared_from_this<DatasetOp> {
  // @return connector size of current op
  int32_t ConnectorCapacity() const {
    if (!inlined()) {
-      return out_connector_->size();
+      return out_connector_->capacity();
    }
    // Return -1 for inlined op
    return -1;