[TensorExpr] Allow for 'keepdim' argument in aten::mean in NNC's external call. (#68756)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/68756
That fixes some warnings in our tests.
Test Plan: Imported from OSS
Reviewed By: navahgar
Differential Revision: D32600952
Pulled By: ZolotukhinM
fbshipit-source-id: 548eaf3659e20795cce44d8f57e77f4a47d44d98
Author
Mikhail Zolotukhin