未验证 提交 2656e90b 编写于 作者: 石晓伟 提交者: GitHub

fix a shape bug in anakin-nvgpu, test=release/1.5 (#19383)

上级 c328a9e5
...@@ -195,7 +195,7 @@ bool PaddleInferenceAnakinPredictor<T, P, R>::RunImpl( ...@@ -195,7 +195,7 @@ bool PaddleInferenceAnakinPredictor<T, P, R>::RunImpl(
<< "'s type is not float"; << "'s type is not float";
} }
auto d_tensor_p = this->executor_p_->get_in(input.name); auto d_tensor_p = this->executor_p_->get_in(input.name);
auto net_shape = d_tensor_p->shape(); auto net_shape = d_tensor_p->valid_shape();
if (net_shape.size() != input.shape.size()) { if (net_shape.size() != input.shape.size()) {
LOG(FATAL) << " input " << input.name LOG(FATAL) << " input " << input.name
<< "'s shape size should be equal to that of net"; << "'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.
先完成此消息的编辑!
想要评论请 注册