From 354d8b0354ac0ea922dfbbbcf17c0c64b4bb6c0e Mon Sep 17 00:00:00 2001
From: changzherui <changzherui1@huawei.com>
Date: Tue, 5 May 2020 16:02:02 +0800
Subject: [PATCH] mod softmax

---
 mindspore/ccsrc/kernel/tbe/tbe_adapter.cc | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/mindspore/ccsrc/kernel/tbe/tbe_adapter.cc b/mindspore/ccsrc/kernel/tbe/tbe_adapter.cc
index fec4729b2..959d21a1b 100644
--- a/mindspore/ccsrc/kernel/tbe/tbe_adapter.cc
+++ b/mindspore/ccsrc/kernel/tbe/tbe_adapter.cc
@@ -30,6 +30,9 @@ namespace mindspore {
 namespace kernel {
 namespace tbe {
 static std::map<string, string> tbe_func_adapter_map = {
+  {"softmax", "softmax_v2"},
+  {"log_softmax", "log_softmax_v2"},
+  {"applymomentum", "applymomentum_d"},
   {"re_lu6", "relu6"},
   {"re_lu6_grad", "relu6_grad"},
   {"re_lu", "relu"},
-- 
GitLab