Replacing axis with dim in quantized cat
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/24151
Test Plan: Imported from OSS
Differential Revision: D16754347
Pulled By: zafartahirov
fbshipit-source-id: c2ebab2f25e0423f16d4f329f98b2e9e138ed549