Add a parameter to BuildHloFromGraph to enable the choice to force the use of `_output_shapes` attribute.
Add a parameter to BuildHloFromGraph to enable the choice to force the use of `_output_shapes` attribute.
