diff options
author | Jerry Ge <jerry.ge@arm.com> | 2024-04-01 17:05:10 +0000 |
---|---|---|
committer | Jerry Ge <jerry.ge@arm.com> | 2024-04-02 21:39:09 +0000 |
commit | 12159fc6fb776908f48fbda9c74cf34980540e4f (patch) | |
tree | 54408e7a10502a347fc7afa25b665c60d696b4d1 /reference_model/src/tensor.cc | |
parent | 9a97eb6cd6aab5eb58eb7860faa9fea305e37c07 (diff) | |
download | reference_model-12159fc6fb776908f48fbda9c74cf34980540e4f.tar.gz |
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
Diffstat (limited to 'reference_model/src/tensor.cc')
-rw-r--r-- | reference_model/src/tensor.cc | 13 |
1 files changed, 11 insertions, 2 deletions
diff --git a/reference_model/src/tensor.cc b/reference_model/src/tensor.cc index 1417fed..088c327 100644 --- a/reference_model/src/tensor.cc +++ b/reference_model/src/tensor.cc @@ -85,8 +85,17 @@ int TosaReference::Tensor::addConsumer(GraphNode* node) int TosaReference::Tensor::dumpTensorParams(FILE* out) const { - fprintf(out, "Name: %s DType=%s isValid=%d Rank=%d Shape=%s\n", tensorName.c_str(), EnumNameTOSAREFTYPE(getDtype()), - getIsValid(), getRank(), getShapeAsString().c_str()); + if (this->getShapeValueSize() > 0) + { + fprintf(out, "Name: %s DType=%s isValid=%d Rank=%d Shape=%s ShapeValue=%s\n", tensorName.c_str(), + EnumNameTOSAREFTYPE(getDtype()), getIsValid(), getRank(), getShapeAsString().c_str(), + getShapeValueAsString().c_str()); + } + else + { + fprintf(out, "Name: %s DType=%s isValid=%d Rank=%d Shape=%s\n", tensorName.c_str(), + EnumNameTOSAREFTYPE(getDtype()), getIsValid(), getRank(), getShapeAsString().c_str()); + } return 0; } |