diff options
author | 2017-06-13 14:30:15 -0700 | |
---|---|---|
committer | 2017-06-13 14:35:34 -0700 | |
commit | ee511ae860c3290db2fb01278fcfff070ee15a16 (patch) | |
tree | 56272272e7a56c4e3337b15b96bfba6d0d38ce1c /tensorflow/core/kernels/quantized_resize_bilinear_op_test.cc | |
parent | d48e0d2e1883be002bec72c2388998c77e61fbee (diff) |
Optimize quantized resize bilinear with NEON-acceleration for ARM devices
PiperOrigin-RevId: 158899240
Diffstat (limited to 'tensorflow/core/kernels/quantized_resize_bilinear_op_test.cc')
-rw-r--r-- | tensorflow/core/kernels/quantized_resize_bilinear_op_test.cc | 10 |
1 files changed, 8 insertions, 2 deletions
diff --git a/tensorflow/core/kernels/quantized_resize_bilinear_op_test.cc b/tensorflow/core/kernels/quantized_resize_bilinear_op_test.cc index f62cb6429a..8d3d7105a4 100644 --- a/tensorflow/core/kernels/quantized_resize_bilinear_op_test.cc +++ b/tensorflow/core/kernels/quantized_resize_bilinear_op_test.cc @@ -332,6 +332,12 @@ void TestResizeBilinearTwoDimsType(const float tolerance, const bool relative) { relative); RunTestResizeBilinearTwoDims<T>(1, 256, 256, 128, 128, 1, tolerance, relative); + RunTestResizeBilinearTwoDims<T>(1, 1, 128, 1, 256, 2, tolerance, relative); + RunTestResizeBilinearTwoDims<T>(1, 128, 1, 256, 1, 2, tolerance, relative); + RunTestResizeBilinearTwoDims<T>(1, 128, 128, 256, 256, 2, tolerance, + relative); + RunTestResizeBilinearTwoDims<T>(1, 256, 256, 128, 128, 2, tolerance, + relative); RunTestResizeBilinearTwoDims<T>(1, 1, 16, 1, 32, 3, tolerance, relative); RunTestResizeBilinearTwoDims<T>(1, 1, 128, 1, 256, 3, tolerance, relative); RunTestResizeBilinearTwoDims<T>(1, 128, 128, 256, 256, 3, tolerance, @@ -352,8 +358,8 @@ void RunBenchmarkResizeBilinearTwoDimsType() { RunBenchmarkResizeBilinearTwoDims<T>(1, 1, 1, 2, 2, 1, ITER); RunBenchmarkResizeBilinearTwoDims<T>(1, 128, 128, 256, 256, 1, ITER); RunBenchmarkResizeBilinearTwoDims<T>(1, 128, 128, 256, 256, 3, ITER); - RunBenchmarkResizeBilinearTwoDims<T>(1, 128 * 2, 128 * 2, 256 * 2, 256 * 2, 3, - ITER); + RunBenchmarkResizeBilinearTwoDims<T>(1, 64, 64, 128, 128, 2, ITER); + RunBenchmarkResizeBilinearTwoDims<T>(1, 32, 32, 64, 64, 16, ITER); } void RunBenchmarkResizeBilinearTwoDims() { |