[jit] Fix Type equality in some cases (#35719)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/35719
Test Plan: Imported from OSS
Differential Revision: D20749422
Pulled By: suo
fbshipit-source-id: 09b697766c1eb3e56f4cf8acc7e854b0981d7991