Browse Source

!104 add RefSwitch & RegMerge

Merge pull request !104 from 陈叶朦/ref_switch_parse
pull/104/MERGE
i-robot Gitee 4 years ago
parent
commit
9cca0f0df7
1 changed files with 2 additions and 0 deletions
  1. +2
    -0
      parser/common/op_map.cc

+ 2
- 0
parser/common/op_map.cc View File

@@ -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},


Loading…
Cancel
Save