Update on "Move optimized target definitions to op_registration.bzl"

^

So we can pull these definitions into codegen.bzl (we can't pull in targets.bzl files).

Differential Revision: [D74741846](https://our.internmc.facebook.com/intern/diff/D74741846/)

[ghstack-poisoned]

Author: lucylq