Sanjoy Das d7634bbfaf Finish migrating {Tf|Platform}GpuId to {Tf|Platform}DeviceId ...

PiperOrigin-RevId: 361252995
Change-Id: I818798fc00efe7b98c35145ce067204d9e023895

2021-03-05 17:15:52 -08:00

..

common

[TF:TRT] Initialize TensorRT plugin registry before deserializing cuda engines.

2020-08-20 12:49:47 -07:00

convert

Finish migrating {Tf|Platform}GpuId to {Tf|Platform}DeviceId

2021-03-05 17:15:52 -08:00

kernels

Finish migrating {Tf|Platform}GpuId to {Tf|Platform}DeviceId

2021-03-05 17:15:52 -08:00

ops

[TF:TRT] Pass the new parameters from TrtGraphConverterV2 to the TF-TRT bridge.

2021-03-01 16:01:56 -08:00

plugin

[TF:TRT] Remove unused target plugin_cast.

2020-08-20 09:30:07 -07:00

segment

[TF:TRT] Add support for per cluster maximum batch size.

2020-10-20 16:49:30 -07:00

stub

…

utils

[TF:TRT] Pass the new parameters from TrtGraphConverterV2 to the TF-TRT bridge.

2021-03-01 16:01:56 -08:00

BUILD

[TF:TRT] Pass the new parameters from TrtGraphConverterV2 to the TF-TRT bridge.

2021-03-01 16:01:56 -08:00

tensorrt_test.cc

[TF:TRT] Add #if to logger_registry.h for consistency.

2020-06-22 19:19:16 -07:00