Commit 6d3cbd2b authored by Smit Hinsu's avatar Smit Hinsu Committed by TensorFlower Gardener
Browse files

Update TensorRT SegmentGraph function to return node pointers 

This is in preparation to share the graph segmentation code between XLA and
TensorRT.

TESTED with existing unit tests

PiperOrigin-RevId: 228654815
parent ebb34298
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