Cherry-pick to earlgrey_1.0.0: [crypto] Fix the KMAC-KDF implementation to return the output in shares.

Cherry-pick to earlgrey_1.0.0: [crypto] Fix the KMAC-KDF implementation to return the output in shares. #5346