!946 Adapt ResizeNearestNeighbor and ApplyAdam

Merge pull request !946 from zhaozhenlong/vm-adapt-resize-nearest
5 years ago · 889b902d38
parent d78b85c8d4 bd9cebb80a
commit 889b902d38
1 changed files with 3 additions and 3 deletions
--- a/mindspore/ccsrc/kernel/tbe/tbe_adapter.cc
+++ b/mindspore/ccsrc/kernel/tbe/tbe_adapter.cc
@ -72,10 +72,10 @@ static std::map<string, string> tbe_func_adapter_map = {
  {"lamb_next_mv_with_decay_v1", "lamb_next_m_v_with_decay_v1"},
  {"lamb_next_mv", "lamb_next_m_v"},
  {"split", "split_d"},
-  {"resize_nearest_neighbor", "resize_nearest_neighbor_d"},
-  {"resize_nearest_neighbor_grad", "resize_nearest_neighbor_grad_d"},
+  {"resize_nearest_neighbor", "resize_nearest_neighbor_v2_d"},
+  {"resize_nearest_neighbor_grad", "resize_nearest_neighbor_v2_grad_d"},
  {"pad", "pad_d"},
-  {"adam", "apply_adam"}};
+  {"adam", "apply_adam_d"}};

 void TbeAdapter::NormalizeFuncName(std::string *func_name) {
  if (func_name == nullptr) {