Browse Source

update getrealop name

tags/v0.3.0-alpha
wangcong 6 years ago
parent
commit
e477e7617b
2 changed files with 2 additions and 2 deletions
  1. +1
    -1
      mindspore/ccsrc/kernel/tbe/tbe_kernel_build.cc
  2. +1
    -1
      mindspore/ccsrc/kernel/tbe/tbe_kernel_build.h

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

@@ -721,7 +721,7 @@ size_t TbeKernelBuild::GetOptionalInput(const mindspore::CNodePtr &cnode, bool i
return (op_info->inputs_ptr().size() + 1 - cnode->inputs().size());
}

std::string TbeKernelBuild::GetRealOpType(const std::string &origin_type) {
std::string TbeKernelBuild::GetRealOpType(std::string &origin_type) {
static std::map<std::string, std::string> buffer_fussion_op_map = {{"DepthwiseConv2dNative", "DepthwiseConv2D"},
{"TensorAdd", "Add"}};
string result = origin_type;


+ 1
- 1
mindspore/ccsrc/kernel/tbe/tbe_kernel_build.h View File

@@ -76,7 +76,7 @@ class TbeKernelBuild {
std::map<const AnfNodePtr, FusionDataType> *spec_data_input);
static bool IsDynamicInput(const CNodePtr &cnode);
static size_t GetOptionalInput(const CNodePtr &cnode, bool is_dynamic_input);
std::string GetRealOpType(const std::string &origin_type);
std::string GetRealOpType(std::string &origin_type);
};

class TbeKernelJsonCreator {


Loading…
Cancel
Save