From 53cee6589e542bd6d7bd8743076fa21ba2877654 Mon Sep 17 00:00:00 2001
From: yanghaitao <yanghaitao1@huawei.com>
Date: Sat, 13 Jun 2020 21:15:11 +0800
Subject: [PATCH] fix profiling capacity

---
 mindspore/ccsrc/dataset/engine/datasetops/dataset_op.h | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mindspore/ccsrc/dataset/engine/datasetops/dataset_op.h b/mindspore/ccsrc/dataset/engine/datasetops/dataset_op.h
index 57ca40c2a0..21b5cbf7ce 100644
--- 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;