Commit a6976d40 authored by Guangda Lai's avatar Guangda Lai Committed by TensorFlower Gardener
Browse files

Improve TF-TRT logging: 

- set the engine name (without scope) before converting the segment to a
  GraaphDef, so we can print it out and know which graph is being operated on.
- format the output of segment nodes in segment.cc

PiperOrigin-RevId: 238043350
parent 46924b2f
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