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

TF-TRT: Run constant folding after layout optimizer, and run the optimizers only once.

Rational: layout optimizer adds Const nodes followed by Reshape nodes, and TF-TRT supports reshape now, so TF should do const folding instead of letting TF-TRT do do that. Also, running TRT optimizer twice is buggy since it'll redo the conversion and try to add TF functions and TRTEngineOps with conflict names to the graph.
PiperOrigin-RevId: 219815498
parent 3a7acc0c
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment