Browse Source

!2283 fix connect size error in device_queue_profiling_xxx.txt

Merge pull request !2283 from yanghaitao/yht_fix_profiling_inline_node
tags/v0.5.0-beta
mindspore-ci-bot Gitee 6 years ago
parent
commit
b32c7192d5
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