Commit adb61543 authored by Yunxing Dai's avatar Yunxing Dai Committed by TensorFlower Gardener
Browse files

Use xla shape as the output of reprensetation_shape function. 

Mostly refactoring the function.

PiperOrigin-RevId: 220325621
parent 136fc44a
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment