Browse Source

add operator diag and diag_part

tags/v0.3.0-alpha
zhaozhenlong chang zherui 5 years ago
parent
commit
1984e4a1ff
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      mindspore/ccsrc/transform/convert.cc

+ 1
- 0
mindspore/ccsrc/transform/convert.cc View File

@@ -190,6 +190,7 @@ const char kNameAtan2[] = "Atan2";
const char kNameApplyRMSProp[] = "ApplyRMSProp"; const char kNameApplyRMSProp[] = "ApplyRMSProp";
const char kNameApplyCenteredRMSProp[] = "ApplyCenteredRMSProp"; const char kNameApplyCenteredRMSProp[] = "ApplyCenteredRMSProp";



// -----------------OpAdapter initialization-------------- // -----------------OpAdapter initialization--------------
std::unordered_map<std::string, OpAdapterDescPtr> &DfGraphConvertor::get_adpt_map() { std::unordered_map<std::string, OpAdapterDescPtr> &DfGraphConvertor::get_adpt_map() {
static std::unordered_map<std::string, OpAdapterDescPtr> adpt_map = { static std::unordered_map<std::string, OpAdapterDescPtr> adpt_map = {


Loading…
Cancel
Save