[TensorExpr] Remove unused classes from TensorExprKernel. (#53283)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/53283
We had `ShapeArg` and `KernelArg` classes, which were wrappers over
`BufferArg` without adding any new functionality on top of what already
existed. This PR removes them and replace their uses with `BufferArg`s
directly.
Test Plan: Imported from OSS
Reviewed By: navahgar
Differential Revision: D26821993
Pulled By: ZolotukhinM
fbshipit-source-id: d1f95ea069b9f38f1d32424464551df2565b3c49
Author
Mikhail Zolotukhin