Optimize alias_analysis node lookup (#54115)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/54115
Test Plan: Imported from OSS
Reviewed By: gmagogsfm
Differential Revision: D27104047
Pulled By: tugsbayasgalan
fbshipit-source-id: 0ef4e78be9ea7081b63ab2303711746bf09653eb
