diff options
Diffstat (limited to 'reference_model/src/verify/verify_utils.h')
-rw-r--r-- | reference_model/src/verify/verify_utils.h | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/reference_model/src/verify/verify_utils.h b/reference_model/src/verify/verify_utils.h index 15d7ba5..24d65b0 100644 --- a/reference_model/src/verify/verify_utils.h +++ b/reference_model/src/verify/verify_utils.h @@ -91,6 +91,12 @@ std::optional<VerifyConfig> parseVerifyConfig(const char* tensorName, const char /// \brief Extract number of total elements int64_t numElements(const std::vector<int32_t>& shape); +/// \brief Convert a flat index to a shape position +std::vector<int32_t> indexToPosition(int64_t index, const std::vector<int32_t>& shape); + +/// \brief A string representing the shape or position +std::string positionToString(const std::vector<int32_t>& pos); + /// \brief Map API data-type to DType DType mapToDType(tosa_datatype_t dataType); |