未验证 提交 a3e641e9 编写于 作者: W whs 提交者: GitHub

Fix infer shape of warpctc op. (#20653)

test=develop
上级 4922eb6d
...@@ -52,7 +52,7 @@ class WarpCTCOp : public framework::OperatorWithKernel { ...@@ -52,7 +52,7 @@ class WarpCTCOp : public framework::OperatorWithKernel {
sequence_width); sequence_width);
// TODO(liuyiqun): it is tricky to set the wrong dimension here. // TODO(liuyiqun): it is tricky to set the wrong dimension here.
ctx->SetOutputDim("Loss", {logits_dims[0], 1}); ctx->SetOutputDim("Loss", {-1, 1});
} }
protected: protected:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册