Migrate to upstream OpKind (#3398)
* Migrate to upstream torch::lazy::OpKind
* Run linter
* Replace ir::OpKind declarations with torch::lazy::OpKind
* Replace some more ir::OpKind to torch::lazy::OpKind
* Run linter again
* Remove unnecessary code
* Run linter again
* Fix merge conflicts
