|
|
|
@ -195,8 +195,7 @@ class ConformerEncoderLayer(nn.Layer):
|
|
|
|
|
x: paddle.Tensor,
|
|
|
|
|
mask: paddle.Tensor,
|
|
|
|
|
pos_emb: paddle.Tensor,
|
|
|
|
|
mask_pad: paddle.
|
|
|
|
|
Tensor, # paddle.ones([0, 0, 0], dtype=paddle.bool)
|
|
|
|
|
mask_pad: paddle.Tensor, #paddle.ones([0, 0, 0],dtype=paddle.bool)
|
|
|
|
|
att_cache: paddle.Tensor, # paddle.zeros([0, 0, 0, 0])
|
|
|
|
|
cnn_cache: paddle.Tensor, # paddle.zeros([0, 0, 0, 0])
|
|
|
|
|
) -> Tuple[paddle.Tensor, paddle.Tensor, paddle.Tensor, paddle.Tensor]:
|
|
|
|
|