add operator diag and diag_part

This commit is contained in:
zhaozhenlong 2020-03-31 09:34:09 +08:00 committed by chang zherui
parent 4c2aa41f1d
commit 1984e4a1ff
1 changed files with 1 additions and 0 deletions

View File

@ -190,6 +190,7 @@ const char kNameAtan2[] = "Atan2";
const char kNameApplyRMSProp[] = "ApplyRMSProp";
const char kNameApplyCenteredRMSProp[] = "ApplyCenteredRMSProp";
// -----------------OpAdapter initialization--------------
std::unordered_map<std::string, OpAdapterDescPtr> &DfGraphConvertor::get_adpt_map() {
static std::unordered_map<std::string, OpAdapterDescPtr> adpt_map = {