Fine-grained codegen customization

[grebe]

In the current state, we only customize codegen for assert and gate ops. A future change I'm working on extends this functionality to most other ops, but it may be useful for customizations to be even more fine-grained, perhaps per-node.