| 379 | |
| 380 | |
| 381 | def multi_batch_compare(func, pipe, pyfunc_pipe): |
| 382 | for it in range(ITERS): |
| 383 | preprocessed_output1, preprocessed_output2 = pipe.run() |
| 384 | out1, out2, out3 = pyfunc_pipe.run() |
| 385 | in1 = [preprocessed_output1.at(i) for i in range(BATCH_SIZE)] |
| 386 | in2 = [preprocessed_output2.at(i) for i in range(BATCH_SIZE)] |
| 387 | pro1, pro2, pro3 = func(in1, in2) |
| 388 | for i in range(BATCH_SIZE): |
| 389 | assert numpy.array_equal(out1.at(i), pro1[i]) |
| 390 | assert numpy.array_equal(out2.at(i), pro2[i]) |
| 391 | assert numpy.array_equal(out3.at(i), pro3[i]) |
| 392 | |
| 393 | |
| 394 | def run_multi_input_multi_output(func, compare, batch=False): |