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

Refactor TF-TRT API: 

- Add a new GraphConverter class to provide general support for offline
  GraphDef/SavedModel conversion. It also supports post-conversion calibration.
  This way we can make a backend specific converter by inheriting this class
  and overwrite the get_rewriter_config() method to provide backend specific
  rewriter options.
- Add a new TrtGraphConverter class (inherited from GraphConverter) for TRT
  conversion.

PiperOrigin-RevId: 232514321
parent f722aee7
Expand all 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