Home
last modified time | relevance | path

Searched refs:IS_TRT_VERSION_GE (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/
Dtrt_shape_optimization_profiles_test.cc75 #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)
Dtrt_shape_optimization_profiles.cc135 #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()
Dtrt_engine_utils.cc82 #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()
Dtrt_shape_optimization_profiles.h59 #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/
Dconvert_nodes.cc125 #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 …]
Dconvert_nodes_test.cc112 #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 …]
Dutils.h74 #define IS_TRT_VERSION_GE(major, minor, patch, build) \ macro
Dutils.cc252 #if IS_TRT_VERSION_GE(6, 0, 0, 0) in GetNumberOfEngineInputs()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/
Dtrt_engine_op_test.cc229 #if IS_TRT_VERSION_GE(6, 0, 0, 0)
Dtrt_engine_op.cc396 #if !IS_TRT_VERSION_GE(6, 0, 0, 0) in TRTEngineOp()
751 #if IS_TRT_VERSION_GE(6, 0, 0, 0) in ExecuteTrtEngine()