Browse Source

add trans rule Placeholder->Data for parser

pull/461/head
medivh-x 4 years ago
parent
commit
cc112f5e23
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      parser/common/op_map.cc

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

@@ -82,6 +82,7 @@ FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY std::map<std::string, std::stri

FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY map<string, string> tensorflow_train_op_map = {
{"BroadcastGradientArgs", BROADCASTGRADIENTARGS},
{"Placeholder", DATA},
{"StopGradient", STOPGRADIENT},
{"ExpandDims", EXPANDDIMS},
{"DestroyTemporaryVariable", DESTROYTEMPORARYVARIABLE},


Loading…
Cancel
Save