9f43ebd68c
tests. Report an error if the native segment for an TRTEngineOp is executed. This ensures that the TRTEngineOp constructed by the bridge is acceptable to TensorRT to catch bugs. Modify TrtModeTestBase to not build static engines for a graph that generates dynamic shaped values. This avoids native segment execution that is not caused by the inconsistency between the bridge and TensorRT. Temporarily allow native segment execution for VGGBlockTest and VGGBlockNCHWTest to workaround b/159459919. PiperOrigin-RevId: 317597344 Change-Id: I6c268c5c912a1fddcffa5d9763399976ddc0299e |
||
|---|---|---|
| .. | ||
| mlir | ||
| tensorrt | ||
| xla | ||
| BUILD | ||
| __init__.py | ||