Sync the declaration and the tests of RegisterFunctionDefForSubgraphs with the definition
Sync the declaration and the tests of RegisterFunctionDefForSubgraphs with the definition

This caused a linking error, which blocks many XLA commits
