Remove the dims checking when the dim is 3 (test=develop)

revert-14786-revert-14782-revert-14398-imperative
Yihua Xu 7 years ago
parent 64e261c6cd
commit ea00270fe8

@ -100,9 +100,6 @@ void eltwise_forward(const framework::ExecutionContext &ctx,
const T *x_data = x->data<T>();
T *y_data = y->mutable_data<T>(ctx.GetPlace());
PADDLE_ENFORCE(x->dims().size() == 2 || x->dims().size() == 4,
"Input dim must be with 2 or 4");
std::vector<int> src_tz = framework::vectorize2int(x->dims());
auto src_format =

Loading…
Cancel
Save