STT-tensorflow/tensorflow/python/compiler
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