diff options
Diffstat (limited to 'runtime/onert/backend/cpu/ops/ArgMinMaxLayer.cc')
| -rw-r--r-- | runtime/onert/backend/cpu/ops/ArgMinMaxLayer.cc | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/runtime/onert/backend/cpu/ops/ArgMinMaxLayer.cc b/runtime/onert/backend/cpu/ops/ArgMinMaxLayer.cc index 2fd284c91..d5ffdef0b 100644 --- a/runtime/onert/backend/cpu/ops/ArgMinMaxLayer.cc +++ b/runtime/onert/backend/cpu/ops/ArgMinMaxLayer.cc @@ -79,6 +79,9 @@ void ArgMinMaxLayer::run() case ir::DataType::UINT8: TF_LITE_ARG_MIN_MAX(uint8_t, int32_t, int32_t); break; + case ir::DataType::QUANT_INT8_ASYMM: + TF_LITE_ARG_MIN_MAX(uint8_t, int32_t, int32_t); + break; case ir::DataType::INT32: TF_LITE_ARG_MIN_MAX(int32_t, int32_t, int32_t); break; @@ -97,6 +100,9 @@ void ArgMinMaxLayer::run() case ir::DataType::UINT8: TF_LITE_ARG_MIN_MAX(uint8_t, int32_t, int64_t); break; + case ir::DataType::QUANT_INT8_ASYMM: + TF_LITE_ARG_MIN_MAX(uint8_t, int32_t, int64_t); + break; case ir::DataType::INT32: TF_LITE_ARG_MIN_MAX(int32_t, int32_t, int64_t); break; |