def model(self): x1 = L.Input(shape=(None, )) x2 = L.Input(shape=(self.hidden_size, )) x3 = L.Input(shape=(self.hidden_size, )) return tf.keras.Model(inputs=[x1, x2, x3], outputs=self.call(x1, x2, x3), name='AttGRUDecoder')
def model(self) -> tf.keras.Model: x1 = L.Input(shape=(None,)) x2 = L.Input(shape=(self.hidden_size,)) return tf.keras.Model(inputs=[x1, x2], outputs=self.call(x1, x2), name='GRUEncoder')