Adjust renamed Orbax V1 API calls.#3220

copybara-service[bot] wants to merge 1 commit into

maingoogle/orbax:mainfrom

test_917950932google/orbax:test_917950932Copy head branch name to clipboard