[C2] Add shape inference logic for ColwiseMax operator. (#51914)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/51914
As desc.
Test Plan: Unit-test.
Reviewed By: intermilan
Differential Revision: D26299115
fbshipit-source-id: 9c80236f843e907476da1747dcd623c85147fa90