diff options
author | Jeremy Johnson <jeremy.johnson@arm.com> | 2024-01-04 17:05:24 +0000 |
---|---|---|
committer | Jeremy Johnson <jeremy.johnson@arm.com> | 2024-01-30 11:49:56 +0000 |
commit | 4f931307a6319d9d99b3afce4ca6e1cd30d77f01 (patch) | |
tree | 5661b63bd087b210403e3b50dbc0ce0a9f8a41b4 /reference_model/src/generate/generate_utils.cc | |
parent | 2d7e4b13d2c3022ae8176d59e2a11d5584ea1d0b (diff) | |
download | reference_model-4f931307a6319d9d99b3afce4ca6e1cd30d77f01.tar.gz |
Main Compliance: DEPTHWISE_CONV2D support
Added DEPTHWISE_CONV2D data generation.
Updated test generation for FP16 and FP32.
Signed-off-by: Jeremy Johnson <jeremy.johnson@arm.com>
Change-Id: I0471d0a1e4e279a27233f4d285082906ceea1bff
Diffstat (limited to 'reference_model/src/generate/generate_utils.cc')
-rw-r--r-- | reference_model/src/generate/generate_utils.cc | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/reference_model/src/generate/generate_utils.cc b/reference_model/src/generate/generate_utils.cc index 917f1b1..b2208c7 100644 --- a/reference_model/src/generate/generate_utils.cc +++ b/reference_model/src/generate/generate_utils.cc @@ -48,6 +48,7 @@ NLOHMANN_JSON_SERIALIZE_ENUM(Op, { Op::Op_CLAMP, "CLAMP" }, { Op::Op_CONCAT, "CONCAT" }, { Op::Op_CONV2D, "CONV2D" }, + { Op::Op_DEPTHWISE_CONV2D, "DEPTHWISE_CONV2D" }, { Op::Op_EQUAL, "EQUAL" }, { Op::Op_ERF, "ERF" }, { Op::Op_EXP, "EXP" }, |