From 993165cfd505e0336cbced09f446beb9cf564f42 Mon Sep 17 00:00:00 2001
From: yanghaitao <yanghaitao1@huawei.com>
Date: Thu, 18 Jun 2020 14:11:26 +0800
Subject: [PATCH] fix profiling size is -1

---
 mindspore/ccsrc/dataset/engine/datasetops/dataset_op.h | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/mindspore/ccsrc/dataset/engine/datasetops/dataset_op.h b/mindspore/ccsrc/dataset/engine/datasetops/dataset_op.h
index 21b5cbf7ce..b47ceea690 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<DatasetOp> {
     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<DatasetOp> {
     if (!inlined()) {
       return out_connector_->capacity();
     }
-    // Return -1 for inlined op
-    return -1;
+    // Return child connector capacity for inlined op
+    return ChildOpConnectorCapacity();
   }
 
   // Getter function