diff --git a/mindspore/ccsrc/kernel/tbe/tbe_adapter.cc b/mindspore/ccsrc/kernel/tbe/tbe_adapter.cc index 8ce5504b8eabd90eaea618c422fef6e1016dd46e..005c290abab7a18a822020deaf11db9f5e87c391 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) {