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

Use TrtNodeValidator to determine whether an op type is supported, and register 

all node converters as validators as well.

This also fix a bug where it says Floor is supported (since it's in the op list
in IsTensorRTCandidate(), and validator_.ValidateNode() returns OK since no
validator was registered for it), but actually it's not supported until TRT
5.1.

PiperOrigin-RevId: 235295283
parent 8af58fde
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