Show actual runtime value of shapeType tensors
* Enable showing actual runtime shapeType tensor value when the
--dump_intermediates=1 flag is on
Signed-off-by: Jerry Ge <jerry.ge@arm.com>
Change-Id: Ibd5aa8aa27505364fbbf9d1addd0bdef0deda885
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 @@
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();
}