Commit 45c5118f authored by Mark Heffernan's avatar Mark Heffernan Committed by TensorFlower Gardener
Browse files

When creating an HloModule from an HloProto construct the HloModuleConfig 

with a correct ProgramShape which matches the shapes of the entry computation.
Previously the module config had a bogus or default constructed ProgramShape.

PiperOrigin-RevId: 173741104
parent 09a89ae5
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