diff options
| author | Jeremy Johnson <jeremy.johnson@arm.com> | 2023-11-15 16:25:45 +0000 |
|---|---|---|
| committer | Eric Kunze <eric.kunze@arm.com> | 2023-11-30 18:52:24 +0000 |
| commit | 708da823504b9a7f4e2ffc10e00f06bb092ce637 (patch) | |
| tree | accbf5aaf055cb07d60fec3c14b7001a8c0fc710 /reference_model | |
| parent | 3047625f7d4b3a77cb3a3480481122f7ba01be2d (diff) | |
| download | reference_model-708da823504b9a7f4e2ffc10e00f06bb092ce637.tar.gz | |
Main Compliance testing support for CAST
Limit CAST input tensor to maximums of output type to avoid
saturation and infinity.
Signed-off-by: Jeremy Johnson <jeremy.johnson@arm.com>
Change-Id: I33350a4ce0ec828da7d2e7aa8cd3183a89a97431
Diffstat (limited to 'reference_model')
| -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 58a3d33..d1758fc 100644 --- a/reference_model/src/generate/generate_utils.cc +++ b/reference_model/src/generate/generate_utils.cc @@ -42,6 +42,7 @@ NLOHMANN_JSON_SERIALIZE_ENUM(Op, { Op::Op_ADD, "ADD" }, { Op::Op_ARGMAX, "ARGMAX" }, { Op::Op_AVG_POOL2D, "AVG_POOL2D" }, + { Op::Op_CAST, "CAST" }, { Op::Op_CEIL, "CEIL" }, { Op::Op_CLAMP, "CLAMP" }, { Op::Op_CONV2D, "CONV2D" }, |