diff --git a/parser/common/op_map.cc b/parser/common/op_map.cc
index 486b462..7efb5c0 100644
--- a/parser/common/op_map.cc
+++ b/parser/common/op_map.cc
@@ -117,7 +117,9 @@ FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY map<string, string> tensorflow_
     {"ShapeN", SHAPEN},
     {"Rank", RANK},
     {"Merge", MERGE},
+    {"RefMerge", REFMERGE},
     {"Switch", SWITCH},
+    {"RefSwitch", REFSWITCH},
     {"LayerNorm", LAYERNORM},
     {"LayerNormGrad", LAYERNORMGRAD},
     {"Dropout", DROPOUT},