Enable batch_size = 0 support in DNNLOWP Concat operator (#26849)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/26849
We were having division-by-zero errors when one of the input tensor dimension is 0 . Examples: P111481720 and P111481374
This diff adds unit tests for empty input tensors and fixes division-by-zero errors in the partition function.
Test Plan: buck test caffe2/caffe2/quantization/server:concat_dnnlowp_op_test -- --stress-runs=100
Reviewed By: jianyuh
Differential Revision: D17574566
fbshipit-source-id: 1d2c21308bde99b3c4f2da82f53201eec42b5d8b