diff options
| author | TatWai Chong <tatwai.chong@arm.com> | 2022-06-08 12:17:14 -0700 |
|---|---|---|
| committer | Eric Kunze <eric.kunze@arm.com> | 2022-08-10 23:12:28 +0000 |
| commit | f732609a51630c98bc3f448937988fbcf20dc854 (patch) | |
| tree | c19aeb8919e4ae8db84b13559bd5858f5e785382 /reference_model/src/graph_node.h | |
| parent | 9391243d1c24d5a4e3c227ba9e3689cdb12d3e60 (diff) | |
| download | reference_model-f732609a51630c98bc3f448937988fbcf20dc854.tar.gz | |
Update TOSA resize to match specification
Attribute stride and shift are removed, and has new scale and border.
Also add tests in the generator to test tf.resize with all option
combinations.
Signed-off-by: TatWai Chong <tatwai.chong@arm.com>
Signed-off-by: Jeremy Johnson <jeremy.johnson@arm.com>
Change-Id: If0f330d04395762d2d907863235eda1532f5e1ff
Diffstat (limited to 'reference_model/src/graph_node.h')
| -rw-r--r-- | reference_model/src/graph_node.h | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/reference_model/src/graph_node.h b/reference_model/src/graph_node.h index 787e89d..adf81b2 100644 --- a/reference_model/src/graph_node.h +++ b/reference_model/src/graph_node.h @@ -37,6 +37,9 @@ #define DEF_INSTANTIATE_TWO_TYPE(OP, DTYPE1, DTYPE2) template class TosaReference::OP<DType_##DTYPE1, DType_##DTYPE2>; +#define DEF_INSTANTIATE_THREE_TYPE(OP, DTYPE1, DTYPE2, OP_TYPE) \ + template class TosaReference::OP<DType_##DTYPE1, DType_##DTYPE2, OP_TYPE>; + #define DEF_INSTANTIATE_RANK0_6_ONE_RANK_ONE_TYPE(OP, DTYPE) \ DEF_INSTANTIATE_ONE_RANK_ONE_TYPE(OP, 0, DTYPE) \ DEF_INSTANTIATE_ONE_RANK_ONE_TYPE(OP, 1, DTYPE) \ @@ -232,6 +235,9 @@ public: return nodeType; } + // Helper functions. + int idiv_check(int input1, int input2, int& result); + protected: // Print out a node validation error int printNodeValidationError(const std::string& msg); |