fix output size adjustment for onnxifi_op

Summary: this breaks if we cut the net at certain int8 ops boundary.

Test Plan: with net_runner to lower a single Int8Quantize op. It used to break. Now it works.

Reviewed By: yinghai

Differential Revision: D22912178

fbshipit-source-id: ca306068c9768df84c1cfa8b34226a1330e19912

caffe2/opt/onnxifi_op.cc

@@ -395,7 +395,7 @@ int OnnxifiOp<CPUContext>::extractOutputBatchSizes() {
           real_shape.dims(j),
           ")");
       begin_ptr[j] = 0;
-      if (max_shape[j] > real_shape.dims(j)) {
+      if (max_shape[j] >= real_shape.dims(j)) {
         end_ptr[j] = real_shape.dims(j);
         mismatch += j;
       } else {