Browse Source

fix profiling size is -1

tags/v0.5.0-beta
yanghaitao 6 years ago
parent
commit
993165cfd5
1 changed files with 4 additions and 4 deletions
  1. +4
    -4
      mindspore/ccsrc/dataset/engine/datasetops/dataset_op.h

+ 4
- 4
mindspore/ccsrc/dataset/engine/datasetops/dataset_op.h View File

@@ -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


Loading…
Cancel
Save