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