diff options
Diffstat (limited to 'reference_model/src/ops/data_layout.cc')
-rw-r--r-- | reference_model/src/ops/data_layout.cc | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/reference_model/src/ops/data_layout.cc b/reference_model/src/ops/data_layout.cc index 4c17e78..e264284 100644 --- a/reference_model/src/ops/data_layout.cc +++ b/reference_model/src/ops/data_layout.cc @@ -270,6 +270,15 @@ int OpDim<Rank, Dtype>::eval() this->out->getTensor().setValues({ out_val }); + // set the shapeValue given the actual tensor value + std::vector<int> shapeValue; + for (int i = 0; i < out->getTensor().size(); ++i) + { + shapeValue.push_back(out->getTensor()(i)); + } + + this->getOutputs()[0]->setShapeValue(shapeValue); + return GraphNode::eval(); } |