diff options
Diffstat (limited to 'reference_model/src/generate/generate_utils.cc')
| -rw-r--r-- | reference_model/src/generate/generate_utils.cc | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/reference_model/src/generate/generate_utils.cc b/reference_model/src/generate/generate_utils.cc index bdc840d..d31048f 100644 --- a/reference_model/src/generate/generate_utils.cc +++ b/reference_model/src/generate/generate_utils.cc @@ -51,6 +51,7 @@ NLOHMANN_JSON_SERIALIZE_ENUM(Op, { Op::Op_EXP, "EXP" }, { Op::Op_FLOOR, "FLOOR" }, { Op::Op_FULLY_CONNECTED, "FULLY_CONNECTED" }, + { Op::Op_GATHER, "GATHER" }, { Op::Op_GREATER, "GREATER" }, { Op::Op_GREATER_EQUAL, "GREATER_EQUAL" }, { Op::Op_IDENTITY, "IDENTITY" }, @@ -69,6 +70,7 @@ NLOHMANN_JSON_SERIALIZE_ENUM(Op, { Op::Op_REDUCE_MAX, "REDUCE_MAX" }, { Op::Op_REDUCE_MIN, "REDUCE_MIN" }, { Op::Op_REDUCE_SUM, "REDUCE_SUM" }, + { Op::Op_SCATTER, "SCATTER" }, { Op::Op_SIGMOID, "SIGMOID" }, { Op::Op_SUB, "SUB" }, { Op::Op_TANH, "TANH" }, |