Home
last modified time | relevance | path

Searched refs:TfTrtIntegrationTestBase (Results 1 – 25 of 26) sorted by relevance

12

/external/tensorflow/tensorflow/python/compiler/tensorrt/test/
Dbase_test.py34 class SimpleSingleEngineTest(trt_test.TfTrtIntegrationTestBase):
74 class SimpleMultiEnginesTest(trt_test.TfTrtIntegrationTestBase):
121 super(trt_test.TfTrtIntegrationTestBase, self).setUp()
127 class SimpleMultiEnginesTest2(trt_test.TfTrtIntegrationTestBase):
168 class ConstInputTest(trt_test.TfTrtIntegrationTestBase):
204 class ConstDataInputSingleEngineTest(trt_test.TfTrtIntegrationTestBase):
225 class ConstDataInputMultipleEnginesTest(trt_test.TfTrtIntegrationTestBase):
257 class ControlDependencyTest(trt_test.TfTrtIntegrationTestBase):
Dint32_test.py32 class ExcludeUnsupportedInt32Test(trt_test.TfTrtIntegrationTestBase):
50 super(trt_test.TfTrtIntegrationTestBase, self).setUp()
66 class CalibrationInt32Support(trt_test.TfTrtIntegrationTestBase):
Dconv2d_test.py83 class Conv2DNCHWTest(trt_test.TfTrtIntegrationTestBase):
118 class Conv2DNHWCTest(trt_test.TfTrtIntegrationTestBase):
141 class Conv2DStridedNCHWTest(trt_test.TfTrtIntegrationTestBase):
172 class Conv2DTranposeTest(trt_test.TfTrtIntegrationTestBase):
Dquantization_test.py56 class QuantizationMissingAllRangesTest(trt_test.TfTrtIntegrationTestBase):
81 class QuantizationWithRangesTest(trt_test.TfTrtIntegrationTestBase):
109 class NonQuantizedPrecisionsWithRangesTest(trt_test.TfTrtIntegrationTestBase):
Dbinary_tensor_weight_broadcast_test.py31 class BinaryTensorWeightBroadcastTest(trt_test.TfTrtIntegrationTestBase):
68 super(trt_test.TfTrtIntegrationTestBase, self).setUp()
Dtopk_test.py29 class TopKTest(trt_test.TfTrtIntegrationTestBase):
50 class TopKOutputTypeTest(trt_test.TfTrtIntegrationTestBase):
Dvgg_block_nchw_test.py35 class VGGBlockNCHWTest(trt_test.TfTrtIntegrationTestBase):
76 super(trt_test.TfTrtIntegrationTestBase, self).setUp()
Dvgg_block_test.py35 class VGGBlockTest(trt_test.TfTrtIntegrationTestBase):
67 super(trt_test.TfTrtIntegrationTestBase, self).setUp()
Ddynamic_input_shapes_test.py32 class DynamicInputShapesTest(trt_test.TfTrtIntegrationTestBase):
84 super(trt_test.TfTrtIntegrationTestBase, self).setUp()
Dreshape_transpose_test.py28 class ReshapeTest(trt_test.TfTrtIntegrationTestBase):
79 class TransposeTest(trt_test.TfTrtIntegrationTestBase):
Dbiasadd_matmul_test.py33 class BiasaddMatMulTest(trt_test.TfTrtIntegrationTestBase):
110 super(trt_test.TfTrtIntegrationTestBase, self).setUp()
Didentity_output_test.py36 class IdentityTest(trt_test.TfTrtIntegrationTestBase):
Dcast_test.py31 class CastInt32ToFp32Test(trt_test.TfTrtIntegrationTestBase):
Dmemory_alignment_test.py31 class MemoryAlignmentTest(trt_test.TfTrtIntegrationTestBase):
Dlru_cache_test.py33 class LRUCacheTest(trt_test.TfTrtIntegrationTestBase):
Dneighboring_engine_test.py32 class NeighboringEngineTest(trt_test.TfTrtIntegrationTestBase):
Dconst_broadcast_test.py28 class ConstBroadcastTest(trt_test.TfTrtIntegrationTestBase):
Dmulti_connection_neighbor_engine_test.py31 class MultiConnectionNeighborEngineTest(trt_test.TfTrtIntegrationTestBase):
Drank_two_test.py30 class RankTwoTest(trt_test.TfTrtIntegrationTestBase):
Dtrt_mode_test.py30 class TrtModeTestBase(trt_test.TfTrtIntegrationTestBase):
Dannotate_max_batch_sizes_test.py31 class MaxBatchSizesTestBase(trt_test.TfTrtIntegrationTestBase):
Dbatch_matmul_test.py34 class BatchMatMultTestBase(trt_test.TfTrtIntegrationTestBase):
Dconcatenation_test.py32 class ConcatenationTest(trt_test.TfTrtIntegrationTestBase):
Dunary_test.py33 class UnaryTest(trt_test.TfTrtIntegrationTestBase):
Dcombined_nms_test.py31 class CombinedNmsTest(trt_test.TfTrtIntegrationTestBase):

12