未验证 提交 8fb760da 编写于 作者: 石晓伟 提交者: GitHub

change anakin apis, test=develop (#20692)

上级 4cf499c0
......@@ -203,7 +203,8 @@ bool PaddleInferenceAnakinPredictor<T, P, R>::RunImpl(
<< "'s type is not float";
}
auto d_tensor_p = this->executor_p_->get_in(input.name);
auto net_shape = d_tensor_p->valid_shape();
// For backward compatibility.
auto net_shape = d_tensor_p->shape();
if (net_shape.size() != input.shape.size()) {
LOG(FATAL) << " input " << input.name
<< "'s shape size should be equal to that of net";
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册