forked from mindspore-Ecosystem/mindspore
fix profiling capacity
This commit is contained in:
parent
f1a004cfa2
commit
53cee6589e
|
@ -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;
|
||||
|
|
Loading…
Reference in New Issue