Ensure backward compatibility with the anakin interface, test=develop (#20691)

* support MLU nums, test=develop

* change anakin apis, test=develop
revert-20712-fix_depthwise_conv
石晓伟 5 years ago committed by GitHub
parent 9a3e22aad4
commit d8f4f4239d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -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";

Loading…
Cancel
Save