Home
last modified time | relevance | path

Searched defs:set_shift (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/core/ops/fusion/
Dpow_fusion.cc34 void PowFusion::set_shift(const float &shift) { (void)this->AddAttr(kShift, api::MakeValue(shift));… in set_shift() function in mindspore::ops::PowFusion
Dexp_fusion.cc40 void ExpFusion::set_shift(const float shift) { (void)this->AddAttr(kShift, api::MakeValue(shift)); } in set_shift() function in mindspore::ops::ExpFusion
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/grad/
Dpower_grad.cc42 void PowerGrad::set_shift(const float shift) { (void)this->AddAttr(kShift, api::MakeValue(shift)); } in set_shift() function in mindspore::ops::PowerGrad
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/auto_generate/
Dgen_lite_ops.cc1833 void Roll::set_shift(const std::vector<int64_t> &shift) { (void)this->AddAttr("shift", … in set_shift() function in mindspore::ops::Roll