diff options
Diffstat (limited to 'scripts/operator_api/templates/operators_cc.j2')
| -rw-r--r-- | scripts/operator_api/templates/operators_cc.j2 | 13 |
1 files changed, 13 insertions, 0 deletions
diff --git a/scripts/operator_api/templates/operators_cc.j2 b/scripts/operator_api/templates/operators_cc.j2 index 6b6f864..1de103a 100644 --- a/scripts/operator_api/templates/operators_cc.j2 +++ b/scripts/operator_api/templates/operators_cc.j2 @@ -92,6 +92,19 @@ tosa::ResizeMode translate_client_tosa_mode(tosa_mode_t mode) { } } +tosa::DType translate_client_acc_size(tosa_acc_size_t acc_size) { + switch(acc_size) { + case tosa_acc_size_int32_t: + return tosa::DType::DType_INT32; + case tosa_acc_size_fp16_t: + return tosa::DType::DType_FP16; + case tosa_acc_size_fp32_t: + return tosa::DType::DType_FP32; + default: + return tosa::DType::DType_UNKNOWN; + } +} + } // namespace extern "C" |