Home
last modified time | relevance | path

Searched defs:set_num_layers (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/core/ops/grad/
Dgru_v2_grad.cc38 void GRUV2Grad::set_num_layers(const int64_t num_layers) { in set_num_layers() function in mindspore::ops::GRUV2Grad
Dlstm_grad_weight.cc52 void LSTMGradWeight::set_num_layers(const int64_t num_layers) { in set_num_layers() function in mindspore::ops::LSTMGradWeight
Dlstm_grad.cc67 void LSTMGrad::set_num_layers(const int64_t num_layers) { in set_num_layers() function in mindspore::ops::LSTMGrad
Dlstm_grad_data.cc55 void LSTMGradData::set_num_layers(const int64_t num_layers) { in set_num_layers() function in mindspore::ops::LSTMGradData
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Dlstm.cc155 void LSTM::set_num_layers(const int64_t num_layers) { in set_num_layers() function in mindspore::ops::LSTM