This website works better with JavaScript.
Home
Issues
Pull Requests
Milestones
AI流水线
Repositories
Datasets
Forum
实训
竞赛
大数据
AI开发
Register
Sign In
Huawei_Technology
/
mindspore
Not watched
Unwatch
Watch all
Watch but not notify
1
Star
0
Fork
0
Code
Releases
13
Wiki
evaluate
Activity
Issues
0
Pull Requests
0
Datasets
Model
Cloudbrain
HPC
Browse Source
!13547
op type adapter
From:
@liubuyu
Reviewed-by: @zhoufeng54,@jjfeing Signed-off-by:
@jjfeing
tags/v1.2.0-rc1
mindspore-ci-bot
Gitee
4 years ago
parent
7d5701c3e9
23ae84a023
commit
c520f3deef
2 changed files
with
6 additions
and
0 deletions
Split View
Diff Options
Show Stats
Download Patch File
Download Diff File
+4
-0
mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_build.cc
+2
-0
mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_utils.h
+ 4
- 0
mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_build.cc
View File
@@ -165,6 +165,10 @@ bool TbeKernelJsonCreator::GenTbeSingleKernelJson(const std::shared_ptr<mindspor
op_info_json[kJFullName] = anf_node->fullname_with_scope();
json_name_ = op_name + "_" + std::to_string(hash_id) + "_" + std::to_string(device_id);
json_info_ = json_str;
auto iter = tbe::opTypeAdapter.find(op_name);
if (iter != tbe::opTypeAdapter.end()) {
op_name = iter->second;
}
op_info_json["Type"] = op_name;
op_info_json["graph_id"] = AnfAlgo::GetGraphId(anf_node.get());
op_info_json[kJKernelName] = json_name_;
+ 2
- 0
mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_utils.h
View File
@@ -33,6 +33,8 @@ namespace kernel {
namespace tbe {
using std::string;
using std::vector;
const std::map<std::string, std::string> opTypeAdapter = {
{"ReLUV2", "ReluV2"}, {"ReLU6", "Relu6"}, {"ReLU6Grad", "Relu6Grad"}, {"ReLUGrad", "ReluGrad"}, {"ReLU", "Relu"}};
class TbeUtils {
public:
Write
Preview
Loading…
Cancel
Save