Browse Source

Pre Merge pull request !461 from 薛鹏/master

pull/461/MERGE
薛鹏 Gitee 3 years ago
parent
commit
3e9e09ff05
No known key found for this signature in database GPG Key ID: 173E9B9CA92EEF8F
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