代码拉取完成,页面将自动刷新
1575
[Feature] LSTM forward use ND in eval mode.
已合并
What type of PR is this?
Uncomment only one
/kind <>
line, hit enter to put that in a new line, and remove leading whitespaces from that line:/kind bug
/kind task
/kind feature
What does this PR do / why do we need it:
LSTM 和 npu_lstm 接口在推理模式下,前向计算的后五个参数格式变为ND格式
Special notes for your reviewers:
影响评估:
训练模式下,接口行为与原先保持一致,没有影响。
推理模式下,离线推理不受影响,在线推理模式下仅改变数据格式,用户实际使用不受影响。