From 558a1452bfe771fb450bee38fd6012f750bb85c7 Mon Sep 17 00:00:00 2001 From: chenyemeng Date: Tue, 24 Nov 2020 17:27:31 +0800 Subject: [PATCH] add refswitch & refmerge --- parser/common/op_map.cc | 2 ++ 1 file changed, 2 insertions(+) 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 tensorflow_ {"ShapeN", SHAPEN}, {"Rank", RANK}, {"Merge", MERGE}, + {"RefMerge", REFMERGE}, {"Switch", SWITCH}, + {"RefSwitch", REFSWITCH}, {"LayerNorm", LAYERNORM}, {"LayerNormGrad", LAYERNORMGRAD}, {"Dropout", DROPOUT},