Searched defs:set_shift (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/core/ops/fusion/ | ||
| D | exp_fusion.cc | 36 void ExpFusion::set_shift(const float shift) { (void)this->AddAttr(kShift, MakeValue(shift)); } in set_shift() function in mindspore::ops::ExpFusion |
| D | pow_fusion.cc | 32 void PowFusion::set_shift(const float &shift) { (void)this->AddAttr(kShift, MakeValue(shift)); } in set_shift() function in mindspore::ops::PowFusion |
| /third_party/mindspore/mindspore/core/ops/grad/ | ||
| D | power_grad.cc | 43 void PowerGrad::set_shift(const float shift) { (void)this->AddAttr(kShift, MakeValue(shift)); } in set_shift() function in mindspore::ops::PowerGrad |