Changing the TF ops used to implement MixedBlockCompression for faster inference speed.
Changing the TF ops used to implement MixedBlockCompression for faster inference speed.

PiperOrigin-RevId: 416327294

google-research/google-research,google-research/google-research,google-research/google-research,google-research/google-research,google-research/google-research,google-research/google-research,google-research/google-research,google-research/google-research,google-research/google-research,google-research/google-research,google-research/google-research