From 708da823504b9a7f4e2ffc10e00f06bb092ce637 Mon Sep 17 00:00:00 2001 From: Jeremy Johnson Date: Wed, 15 Nov 2023 16:25:45 +0000 Subject: 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 Change-Id: I33350a4ce0ec828da7d2e7aa8cd3183a89a97431 --- reference_model/src/generate/generate_utils.cc | 1 + 1 file changed, 1 insertion(+) (limited to 'reference_model/src/generate') 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" }, -- cgit v1.2.1