Searched refs:kKeepProb (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/core/ops/grad/ |
| D | dropout_grad.cc | 30 …CheckAndConvertUtils::CheckInRange<float>(kKeepProb, keep_prob, kIncludeRight, {0.0, 1.0}, this->n… in set_keep_prob()
31 (void)this->AddAttr(kKeepProb, MakeValue(keep_prob)); in set_keep_prob()
35 auto value_ptr = GetAttr(kKeepProb); in get_keep_prob()
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | dropout.cc | 30 …CheckAndConvertUtils::CheckInRange<float>(kKeepProb, keep_prob, kIncludeRight, {0.0, 1.0}, this->n… in set_keep_prob()
31 (void)this->AddAttr(kKeepProb, MakeValue(keep_prob)); in set_keep_prob()
35 auto value_ptr = this->GetAttr(kKeepProb); in get_keep_prob()
|
| D | broadcast.cc | 30 void Broadcast::set_root_rank(const int64_t root_rank) { (void)this->AddAttr(kKeepProb, MakeValue(r… in set_root_rank()
|
| D | op_utils.h | 105 constexpr auto kKeepProb = "keep_prob"; variable
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/mindir/ |
| D | dropout_unify_mindir.cc | 35 constexpr auto kKeepProb = "keep_prob"; variable
70 if (!AnfAlgo::HasNodeAttr(kKeepProb, cnode)) { in CreateKeepPorbValueNode()
78 auto keep_prob = AnfAlgo::GetNodeAttr<float>(node, kKeepProb); in CreateKeepPorbValueNode()
|