minor updates to rotom mnist test#3031

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

maingoogle/heir:mainfrom

test_926801699google/heir:test_926801699Copy head branch name to clipboard