diff options
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 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" }, |