From bd9cebb80a2726a306ec5b85a5cf872540ab9207 Mon Sep 17 00:00:00 2001 From: zhaozhenlong Date: Wed, 6 May 2020 18:25:35 +0800 Subject: [PATCH] adapt ResizeNearestNeighbourV2 with grad and ApplyAdam --- mindspore/ccsrc/kernel/tbe/tbe_adapter.cc | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/mindspore/ccsrc/kernel/tbe/tbe_adapter.cc b/mindspore/ccsrc/kernel/tbe/tbe_adapter.cc index 8ce5504b8e..005c290aba 100644 --- a/mindspore/ccsrc/kernel/tbe/tbe_adapter.cc +++ b/mindspore/ccsrc/kernel/tbe/tbe_adapter.cc @@ -72,10 +72,10 @@ static std::map 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) {