Browse Source

fix compile error

tags/v0.3.0-alpha
wangcong 5 years ago
parent
commit
6478015786
2 changed files with 3 additions and 2 deletions
  1. +2
    -2
      mindspore/ccsrc/kernel/tbe/tbe_kernel_build.cc
  2. +1
    -0
      mindspore/ccsrc/operator/ops.h

+ 2
- 2
mindspore/ccsrc/kernel/tbe/tbe_kernel_build.cc View File

@@ -722,8 +722,8 @@ size_t TbeKernelBuild::GetOptionalInput(const mindspore::CNodePtr &cnode, bool i
}

std::string TbeKernelBuild::GetRealOpType(const std::string &origin_type) {
static std::map<std::string, std::string> TbeKernelBuild::buffer_fussion_op_map = {
{"DepthwiseConv2dNative", "DepthwiseConv2D"}, {"TensorAdd", "Add"}};
static std::map<std::string, std::string> buffer_fussion_op_map = {{"DepthwiseConv2dNative", "DepthwiseConv2D"},
{"TensorAdd", "Add"}};
string result = origin_type;
if (buffer_fussion_op_map.find(origin_type) != buffer_fussion_op_map.end()) {
result = buffer_fussion_op_map[origin_type];


+ 1
- 0
mindspore/ccsrc/operator/ops.h View File

@@ -204,6 +204,7 @@ extern const PrimitivePtr kPrimOneHot;
extern const PrimitivePtr kPrimGelu;
extern const PrimitivePtr kPrimGeluGrad;
extern const PrimitivePtr kPrimRelu;
extern const PrimitivePtr kPrimReluV2;
extern const PrimitivePtr kPrimActivation;
extern const PrimitivePtr kPrimZerosLikeTensor;
extern const PrimitivePtr kPrimFakeBprop;


Loading…
Cancel
Save