Bixia Zheng 9f43ebd68c [TF:TRT] Enforce no native segment execution for TfTrtIntegrationTestBase ...

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

2020-06-21 23:22:26 -07:00

..

mlir

Export the MLIR classes and functions from C++ to Python with pybind11 instead of swig. This is part of a larger effort to deprecate swig and eventually with modularization break pywrap_tensorflow into smaller components. It will also make exporting C++ ops to Python significantly easier. Please refer to https://github.com/tensorflow/community/blob/master/rfcs/20190208-pybind11.md for more information.

2020-01-28 21:57:23 -08:00

tensorrt

[TF:TRT] Enforce no native segment execution for TfTrtIntegrationTestBase

2020-06-21 23:22:26 -07:00

xla

Allow returning attr types from TPUStrategy.run

2020-06-11 10:38:35 -07:00

__init__.py

Link TF-TRT into python:no_contrib (so it's included in TF 2.0 build) and lazily load TRT shared library.

2019-02-11 15:53:03 -08:00

BUILD

Export tftrt python symbols.

2019-11-07 14:07:32 -08:00