[NNC] Optimize conditionals that correspond to the form generated for aten::cat op. (#57673)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/57673
Test Plan: Imported from OSS
Reviewed By: bertmaher
Differential Revision: D28231374
Pulled By: navahgar
fbshipit-source-id: 1777a63df4e5ebed6d515683bd772a88be465b3a