diff options
Diffstat (limited to 'scripts/py/templates/TestData.hpp.template')
| -rw-r--r-- | scripts/py/templates/TestData.hpp.template | 23 |
1 files changed, 17 insertions, 6 deletions
diff --git a/scripts/py/templates/TestData.hpp.template b/scripts/py/templates/TestData.hpp.template index 13c410d..bd56f9f 100644 --- a/scripts/py/templates/TestData.hpp.template +++ b/scripts/py/templates/TestData.hpp.template @@ -25,13 +25,13 @@ namespace {{namespace}} { {% endfor %} -#define NUMBER_OF_IFM_FILES ({{ifm_count}}U) -#define NUMBER_OF_OFM_FILES ({{ofm_count}}U) +#define NUMBER_OF_IFM_FILES ({{ifm_count}}U) +#define NUMBER_OF_OFM_FILES ({{ofm_count}}U) {% for ifm_size in ifm_var_sizes %} -#define IFM_{{loop.index0}}_DATA_SIZE ({{ifm_size}}U) +#define IFM_{{loop.index0}}_DATA_SIZE ({{ifm_size}}U) {% endfor %} {% for ofm_size in ofm_var_sizes %} -#define OFM_{{loop.index0}}_DATA_SIZE ({{ofm_size}}U) +#define OFM_{{loop.index0}}_DATA_SIZE ({{ofm_size}}U) {% endfor %} {% for ifm_var_name in ifm_var_names %} @@ -42,8 +42,19 @@ extern const {{data_type}} {{ifm_var_name}}[IFM_{{loop.index0}}_DATA_SIZE]; extern const {{data_type}} {{ofm_var_name}}[OFM_{{loop.index0}}_DATA_SIZE]; {% endfor %} -const {{data_type}}* get_ifm_data_array(const uint32_t idx); -const {{data_type}}* get_ofm_data_array(const uint32_t idx); +/** + * @brief Gets the pointer to input feature map (IFM). + * @param[in] idx Index of the input. + * @return Pointer to IFM. + **/ +const {{data_type}}* GetIfmDataArray(const uint32_t idx); + +/** + * @brief Gets the pointer to output feature map (OFM). + * @param[in] idx Index of the output. + * @return Pointer to OFM. + **/ +const {{data_type}}* GetOfmDataArray(const uint32_t idx); {% for namespace in namespaces %} } /* namespace {{namespace}} */ |