/external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/ |
D | trt_shape_optimization_profiles_test.cc | 75 #if IS_TRT_VERSION_GE(6, 0, 0, 0) in SetUp() 111 #if IS_TRT_VERSION_GE(6, 0, 0, 0) 118 #if IS_TRT_VERSION_GE(6, 0, 0, 0) 132 #if IS_TRT_VERSION_GE(6, 0, 0, 0) in TEST_F() 155 #if IS_TRT_VERSION_GE(6, 0, 0, 0)
|
D | trt_shape_optimization_profiles.cc | 135 #if IS_TRT_VERSION_GE(6, 0, 0, 0) 177 #if IS_TRT_VERSION_GE(6, 0, 0, 0) 191 #if IS_TRT_VERSION_GE(6, 0, 0, 0) in SetShapeTensorMask() 253 #if IS_TRT_VERSION_GE(6, 0, 0, 0) in CreateExecutionContexts() 272 #if IS_TRT_VERSION_GE(6, 0, 0, 0) in RestoreProfiles() 277 #if IS_TRT_VERSION_GE(7, 0, 0, 0) in RestoreProfiles()
|
D | trt_engine_utils.cc | 82 #if IS_TRT_VERSION_GE(6, 0, 0, 0) in GetTrtBindingShape() 120 #if IS_TRT_VERSION_GE(6, 0, 0, 0) in GetTrtBindingIndex() 160 #if IS_TRT_VERSION_GE(6, 0, 0, 0) in SetTrtEngineInputs() 202 #if IS_TRT_VERSION_GE(6, 0, 0, 0) in SetTrtEngineInputs() 287 #if IS_TRT_VERSION_GE(6, 0, 0, 0) in TrtEnqueue()
|
D | trt_shape_optimization_profiles.h | 59 #if IS_TRT_VERSION_GE(6, 0, 0, 0) 153 #if IS_TRT_VERSION_GE(6, 0, 0, 0) 201 #if IS_TRT_VERSION_GE(6, 0, 0, 0)
|
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
D | convert_nodes.cc | 125 #if IS_TRT_VERSION_GE(6, 0, 0, 0) in LayerTypeToString() 130 #if IS_TRT_VERSION_GE(7, 0, 0, 0) in LayerTypeToString() 479 #if !IS_TRT_VERSION_GE(5, 1, 3, 0) in CreateConstantLayer() 745 #if IS_TRT_VERSION_GE(7, 0, 0, 0) in size_bytes() 800 #if IS_TRT_VERSION_GE(5, 0, 0, 0) 806 #if IS_TRT_VERSION_GE(5, 1, 0, 0) 816 #if IS_TRT_VERSION_GE(6, 0, 0, 0) 1270 #if IS_TRT_VERSION_GE(6, 0, 0, 0) in Init() 1417 #if IS_TRT_VERSION_GE(7, 1, 3, 0) 1463 #if IS_TRT_VERSION_GE(6, 0, 0, 0) in BuildCudaEngine() [all …]
|
D | convert_nodes_test.cc | 112 #if IS_TRT_VERSION_GE(6, 0, 0, 0) 363 #if IS_TRT_VERSION_GE(5, 0, 0, 0) 372 #if IS_TRT_VERSION_GE(5, 1, 0, 0) 382 #if IS_TRT_VERSION_GE(6, 0, 0, 0) 1109 #if IS_TRT_VERSION_GE(5, 0, 0, 0) in TEST_F() 2434 #if !IS_TRT_VERSION_GE(7, 1, 3, 0) in TEST_P() 3129 #if IS_TRT_VERSION_GE(7, 1, 3, 0) 3682 #if IS_TRT_VERSION_GE(5, 1, 3, 1) in TEST_F() 4005 #if IS_TRT_VERSION_GE(5, 1, 3, 1) in TEST_F() 4113 #if IS_TRT_VERSION_GE(5, 1, 3, 1) in TEST_F() [all …]
|
D | utils.h | 74 #define IS_TRT_VERSION_GE(major, minor, patch, build) \ macro
|
D | utils.cc | 252 #if IS_TRT_VERSION_GE(6, 0, 0, 0) in GetNumberOfEngineInputs()
|
/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
D | trt_engine_op_test.cc | 229 #if IS_TRT_VERSION_GE(6, 0, 0, 0)
|
D | trt_engine_op.cc | 396 #if !IS_TRT_VERSION_GE(6, 0, 0, 0) in TRTEngineOp() 751 #if IS_TRT_VERSION_GE(6, 0, 0, 0) in ExecuteTrtEngine()
|