(self)
| 96 | return decoded, hidden |
| 97 | |
| 98 | def parameter_rrefs(self): |
| 99 | remote_params = [] |
| 100 | # get RRefs of embedding table |
| 101 | remote_params.extend(_remote_method(_parameter_rrefs, self.emb_table_rref)) |
| 102 | # create RRefs for local parameters |
| 103 | remote_params.extend(_parameter_rrefs(self.rnn)) |
| 104 | # get RRefs of decoder |
| 105 | remote_params.extend(_remote_method(_parameter_rrefs, self.decoder_rref)) |
| 106 | return remote_params |
no test coverage detected