Searched refs:kDropout (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/core/ops/grad/ |
| D | lstm_grad.cc | 55 …CheckAndConvertUtils::CheckInRange<float>(kDropout, dropout, kIncludeBoth, {0.0, 1.0}, this->name(… in set_dropout()
56 (void)AddAttr(kDropout, MakeValue(dropout)); in set_dropout()
59 auto value_ptr = this->GetAttr(kDropout); in get_dropout()
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | lstm.cc | 127 …CheckAndConvertUtils::CheckInRange<float>(kDropout, dropout, kIncludeBoth, {0.0, 1.0}, this->name(… in set_dropout()
128 (void)AddAttr(kDropout, MakeValue(dropout)); in set_dropout()
131 auto value_ptr = this->GetAttr(kDropout); in get_dropout()
|
| D | op_utils.h | 66 constexpr auto kDropout = "dropout"; variable
|
| /third_party/mindspore/mindspore/core/base/ |
| D | core_ops.h | 100 constexpr auto kDropout = "Dropout"; variable
359 inline const PrimitivePtr kPrimDropout = std::make_shared<Primitive>(kDropout);
|