Add missing load statements (#4508)

This is required for compatibility with bazel @ HEAD. Done with:
`buildifier --lint=fix -r .`
19 files changed