Searched refs:flatten_parameters (Results 1 – 10 of 10) sorted by relevance
400 gru->flatten_parameters(); in TEST_F()570 gru_cpu->flatten_parameters(); in TEST_F()571 gru_cuda->flatten_parameters(); in TEST_F()625 lstm_cpu->flatten_parameters(); in TEST_F()626 lstm_cuda->flatten_parameters(); in TEST_F()682 lstm_cpu->flatten_parameters(); in TEST_F()683 lstm_cuda->flatten_parameters(); in TEST_F()
175 this->flatten_parameters(); in reset()180 void RNNImplBase<Derived>::flatten_parameters() { in flatten_parameters() function in torch::nn::detail::RNNImplBase262 flatten_parameters(); in to()269 flatten_parameters(); in to()276 flatten_parameters(); in to()
204 def flatten_parameters(self): member in BatchRNN205 self.rnn.flatten_parameters()
52 void flatten_parameters();
89 self.gru.flatten_parameters()
87 self.gru.flatten_parameters()
215 self.flatten_parameters()224 def flatten_parameters(self) -> None: member in RNNBase
202 self.lstm.flatten_parameters()
76 self.rnn.flatten_parameters()
4336 rnn2.flatten_parameters()