diff options
Diffstat (limited to 'scripts/operator_api/templates/operators_h.j2')
-rw-r--r-- | scripts/operator_api/templates/operators_h.j2 | 37 |
1 files changed, 5 insertions, 32 deletions
diff --git a/scripts/operator_api/templates/operators_h.j2 b/scripts/operator_api/templates/operators_h.j2 index 042d7a5..0c98da8 100644 --- a/scripts/operator_api/templates/operators_h.j2 +++ b/scripts/operator_api/templates/operators_h.j2 @@ -21,6 +21,7 @@ #include "func_config.h" #include "func_debug.h" +#include "types.h" #include <stddef.h> #include <stdint.h> @@ -29,37 +30,10 @@ extern "C" { #endif /* __cplusplus */ - // Note status needs to be aligned with graph_status - enum tosa_status_t + struct func_ctx_t { - tosa_status_valid = 0, - tosa_status_unpredictable = 1, - tosa_status_error = 2 - }; - - enum tosa_mode_t - { - tosa_mode_unknown = 0, - tosa_mode_nearest = 1, - tosa_mode_bilinear = 2, - tosa_mode_min = 3, - tosa_mode_max = 4 - }; - - enum tosa_datatype_t - { - {% for dataType in dataTypes: -%} - {{dataType}} = {{loop.index-1}}, - {% endfor -%} - }; - - struct tosa_tensor_t - { - int32_t* shape; - int32_t num_dims; - tosa_datatype_t data_type; - uint8_t* data; - size_t size; + func_config_t func_config = func_config_t{}; + func_debug_t func_debug = func_debug_t{}; }; {% for operator in operators: %} @@ -67,8 +41,7 @@ extern "C" { {%- for arg in operator.arguments: -%} {% if arg.type != "tosa_tensor_t" -%}const {% endif -%}{{arg.type}} client_{{arg.name}}{{arg.shape}} {% if loop.index < operator.arguments|length %},{% endif %} - {%- endfor -%}, const func_config_t &func_config = func_config_t{}, - const func_debug_t &func_debug = func_debug_t{}); + {%- endfor -%},const func_ctx_t& func_ctx); {% endfor %} #ifdef __cplusplus |