Browse Source

add refswitch & refmerge

pull/104/head
chenyemeng 5 years ago
parent
commit
558a1452bf
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