Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/compiler/tensorrt/test/
Dbase_test.py34 class SimpleSingleEngineTest(trt_test.TfTrtIntegrationTestBase):
84 class SimpleMultiEnginesTest(trt_test.TfTrtIntegrationTestBase):
150 class SimpleMultiEnginesTest2(trt_test.TfTrtIntegrationTestBase):
199 class ConstInputTest(trt_test.TfTrtIntegrationTestBase):
244 class ConstDataInputSingleEngineTest(trt_test.TfTrtIntegrationTestBase):
274 class ConstDataInputMultipleEnginesTest(trt_test.TfTrtIntegrationTestBase):
315 class ControlDependencyTest(trt_test.TfTrtIntegrationTestBase):
Dquantization_test.py68 class QuantizationMissingAllRangesTest(trt_test.TfTrtIntegrationTestBase):
92 class QuantizationWithRangesTest(trt_test.TfTrtIntegrationTestBase):
118 class NonQuantizedPrecisionsWithRangesTest(trt_test.TfTrtIntegrationTestBase):
Dconv2d_test.py91 class Conv2DNCHWTest(trt_test.TfTrtIntegrationTestBase):
116 class Conv2DNHWCTest(trt_test.TfTrtIntegrationTestBase):
141 class Conv2DStridedNCHWTest(trt_test.TfTrtIntegrationTestBase):
191 class Conv2DTranposeTest(trt_test.TfTrtIntegrationTestBase):
Dtopk_test.py30 class TopKTest(trt_test.TfTrtIntegrationTestBase):
57 class TopKOutputTypeTest(trt_test.TfTrtIntegrationTestBase):
Dreshape_transpose_test.py29 class ReshapeTest(trt_test.TfTrtIntegrationTestBase):
92 class TransposeTest(trt_test.TfTrtIntegrationTestBase):
Dtf_trt_integration_test_base.py110 class TfTrtIntegrationTestBase(test_util.TensorFlowTestCase): class
153 super(TfTrtIntegrationTestBase, self).__init__(methodName)
158 super(TfTrtIntegrationTestBase, self).setUp()
603 _AddTests(TfTrtIntegrationTestBase)
Didentity_output_test.py37 class IdentityTest(trt_test.TfTrtIntegrationTestBase):
Dbinary_tensor_weight_broadcast_test.py32 class BinaryTensorWeightBroadcastTest(trt_test.TfTrtIntegrationTestBase):
Dmemory_alignment_test.py32 class MemoryAlignmentTest(trt_test.TfTrtIntegrationTestBase):
Dneighboring_engine_test.py33 class NeighboringEngineTest(trt_test.TfTrtIntegrationTestBase):
Dlru_cache_test.py33 class LRUCacheTest(trt_test.TfTrtIntegrationTestBase):
Dint32_test.py33 class ExcludeUnsupportedInt32Test(trt_test.TfTrtIntegrationTestBase):
Dvgg_block_test.py34 class VGGBlockTest(trt_test.TfTrtIntegrationTestBase):
Dconst_broadcast_test.py32 class ConstBroadcastTest(trt_test.TfTrtIntegrationTestBase):
Dvgg_block_nchw_test.py34 class VGGBlockNCHWTest(trt_test.TfTrtIntegrationTestBase):
Dmulti_connection_neighbor_engine_test.py32 class MultiConnectionNeighborEngineTest(trt_test.TfTrtIntegrationTestBase):
Drank_two_test.py31 class RankTwoTest(trt_test.TfTrtIntegrationTestBase):
Ddynamic_input_shapes_test.py32 class DynamicInputShapesTest(trt_test.TfTrtIntegrationTestBase):
Dbatch_matmul_test.py33 class BatchMatMulTest(trt_test.TfTrtIntegrationTestBase):
Dconcatenation_test.py33 class ConcatenationTest(trt_test.TfTrtIntegrationTestBase):
Dunary_test.py34 class UnaryTest(trt_test.TfTrtIntegrationTestBase):
Dbiasadd_matmul_test.py34 class BiasaddMatMulTest(trt_test.TfTrtIntegrationTestBase):