Commit c2aa6611 authored by Skye Wanderman-Milne's avatar Skye Wanderman-Milne Committed by TensorFlower Gardener
Browse files

Fetch shape information from the C API when enabled. 

This change makes set_shapes_for_outputs fetch the already-computed tensor shapes from the C API, rather than calling the C++ shape function. It also moves the set_shapes_for_outputs call so it works with Operations created from TF_Operations.

PiperOrigin-RevId: 177088786
parent c17459a0
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