Searched refs:kHidden_size (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/core/ops/grad/ |
| D | lstm_grad.cc | 40 …(void)CheckAndConvertUtils::CheckInteger(kHidden_size, hidden_size, kGreaterThan, 0, this->name()); in set_hidden_size()
41 (void)AddAttr(kHidden_size, MakeValue(hidden_size)); in set_hidden_size()
43 int64_t LSTMGrad::get_hidden_size() const { return GetValue<int64_t>(GetAttr(kHidden_size)); } in get_hidden_size()
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | lstm.cc | 56 int64_t hidden_size = GetValue<int64_t>(primitive->GetAttr(kHidden_size)); in LstmInfer()
112 …(void)CheckAndConvertUtils::CheckInteger(kHidden_size, hidden_size, kGreaterThan, 0, this->name()); in set_hidden_size()
113 (void)AddAttr(kHidden_size, MakeValue(hidden_size)); in set_hidden_size()
115 int64_t LSTM::get_hidden_size() const { return GetValue<int64_t>(GetAttr(kHidden_size)); } in get_hidden_size()
|
| D | op_utils.h | 215 constexpr auto kHidden_size = "hidden_size"; variable
|