For TF2, TRTEngineOp nodes should only exist inside functions for signatures. Avoid looking for TRTEngineOp nodes outside functions. PiperOrigin-RevId: 358194388 Change-Id: I123b6e366a130c87b1e0f5bc64336bf21f715b20 |
||
---|---|---|
.. | ||
mlir | ||
tensorrt | ||
xla | ||
__init__.py | ||
BUILD |