Shape related code updated to address TODO(b/208879020)#270

Open

pritamdodeja wants to merge 3 commits intotensorflow:masterfrom pritamdodeja:shape_fix

+17-17

Commits on May 6, 2022

Fix to only transform raw data when requested.
pritamdodeja
committed

Commits on May 10, 2022