Searched defs:get_num_layers (Results 1 – 6 of 6) sorted by relevance
50 int64_t get_num_layers() const { return GetValue<int64_t>(GetAttr(kNumLayers)); } in get_num_layers() function
160 int64_t LSTM::get_num_layers() const { return GetValue<int64_t>(GetAttr(kNumLayers)); } in get_num_layers() function in mindspore::ops::LSTM
42 int64_t GRUV2Grad::get_num_layers() const { return GetValue<int64_t>(GetAttr(kNumLayers)); } in get_num_layers() function in mindspore::ops::GRUV2Grad
56 int64_t LSTMGradWeight::get_num_layers() const { return GetValue<int64_t>(GetAttr(kNumLayers)); } in get_num_layers() function in mindspore::ops::LSTMGradWeight
71 int64_t LSTMGrad::get_num_layers() const { return GetValue<int64_t>(GetAttr(kNumLayers)); } in get_num_layers() function in mindspore::ops::LSTMGrad
59 int64_t LSTMGradData::get_num_layers() const { return GetValue<int64_t>(GetAttr(kNumLayers)); } in get_num_layers() function in mindspore::ops::LSTMGradData