Skip to content

M
mindspore

magicwindyyd / mindspore
与 Fork 源项目一致

Fork自 MindSpore / mindspore

通知 1
Star 1
Fork 0
M
mindspore
提交 64780157 编写于 作者: W wangcong
浏览文件
操作

fix compile error

上级 7017ac3f
隐藏空白更改
内联 并排
Showing with 3 addition and 2 deletion
mindspore/ccsrc/kernel/tbe/tbe_kernel_build.cc
浏览文件 @ 64780157
...@@ -722,8 +722,8 @@ size_t TbeKernelBuild::GetOptionalInput(const mindspore::CNodePtr &cnode, bool i ...@@ -722,8 +722,8 @@ size_t TbeKernelBuild::GetOptionalInput(const mindspore::CNodePtr &cnode, bool i
} }
std::string TbeKernelBuild::GetRealOpType(const std::string &origin_type) { std::string TbeKernelBuild::GetRealOpType(const std::string &origin_type) {
static std::map<std::string, std::string> TbeKernelBuild::buffer_fussion_op_map = { static std::map<std::string, std::string> buffer_fussion_op_map = {{"DepthwiseConv2dNative", "DepthwiseConv2D"},
{"DepthwiseConv2dNative", "DepthwiseConv2D"}, {"TensorAdd", "Add"}}; {"TensorAdd", "Add"}};
string result = origin_type; string result = origin_type;
if (buffer_fussion_op_map.find(origin_type) != buffer_fussion_op_map.end()) { if (buffer_fussion_op_map.find(origin_type) != buffer_fussion_op_map.end()) {
result = buffer_fussion_op_map[origin_type]; result = buffer_fussion_op_map[origin_type];
......
mindspore/ccsrc/operator/ops.h
浏览文件 @ 64780157
...@@ -204,6 +204,7 @@ extern const PrimitivePtr kPrimOneHot; ...@@ -204,6 +204,7 @@ extern const PrimitivePtr kPrimOneHot;
extern const PrimitivePtr kPrimGelu; extern const PrimitivePtr kPrimGelu;
extern const PrimitivePtr kPrimGeluGrad; extern const PrimitivePtr kPrimGeluGrad;
extern const PrimitivePtr kPrimRelu; extern const PrimitivePtr kPrimRelu;
extern const PrimitivePtr kPrimReluV2;
extern const PrimitivePtr kPrimActivation; extern const PrimitivePtr kPrimActivation;
extern const PrimitivePtr kPrimZerosLikeTensor; extern const PrimitivePtr kPrimZerosLikeTensor;
extern const PrimitivePtr kPrimFakeBprop; extern const PrimitivePtr kPrimFakeBprop;
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或