提交 edfec1d3 编写于 作者: myq406450149's avatar myq406450149

add lod_rank_table. test=develop

上级 1bd1056f
...@@ -58,7 +58,9 @@ output operators. ...@@ -58,7 +58,9 @@ output operators.
class LoDRankTableInferShape : public framework::InferShapeBase { class LoDRankTableInferShape : public framework::InferShapeBase {
public: public:
void operator()(framework::InferShapeContext *context) const override { void operator()(framework::InferShapeContext *context) const override {
PADDLE_ENFORCE(context->HasInput("X"), "LoDRankTable must have input X"); PADDLE_ENFORCE_EQ(
context->HasInput("X"), true,
platform::errors::NotFound("LoDRankTable must have input X."));
} }
}; };
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册