예제 #1
0
 def _set_input_buffer(self, incremental_state, buffer):
     set_incremental_state(
         self,
         incremental_state,
         'attn_state',
         buffer,
     )
예제 #2
0
 def _set_input_buffer(self, incremental_state, new_buffer):
     return utils.set_incremental_state(self, incremental_state,
                                        'input_buffer', new_buffer)