HomeSort by relevance Sort by last modified time
    Searched refs:trt_test (Results 1 - 21 of 21) sorted by null

  /external/tensorflow/tensorflow/python/compiler/tensorrt/test/
quantization_test.py 23 from tensorflow.python.compiler.tensorrt.test import tf_trt_integration_test_base as trt_test
60 return trt_test.TfTrtIntegrationTestParams(
68 class QuantizationMissingAllRangesTest(trt_test.TfTrtIntegrationTestBase):
78 return (trt_test.IsQuantizationMode(run_params.precision_mode) and
92 class QuantizationWithRangesTest(trt_test.TfTrtIntegrationTestBase):
102 return (trt_test.IsQuantizationMode(run_params.precision_mode) and
118 class NonQuantizedPrecisionsWithRangesTest(trt_test.TfTrtIntegrationTestBase):
126 return not trt_test.IsQuantizationMode(run_params.precision_mode)
dynamic_input_shapes_test.py 23 from tensorflow.python.compiler.tensorrt.test import tf_trt_integration_test_base as trt_test
32 class DynamicInputShapesTest(trt_test.TfTrtIntegrationTestBase):
73 return trt_test.TfTrtIntegrationTestParams(
88 rewriter_config=trt_test.OptimizerDisabledRewriterConfig())
95 not trt_test.IsQuantizationMode(run_params.precision_mode))
int32_test.py 23 from tensorflow.python.compiler.tensorrt.test import tf_trt_integration_test_base as trt_test
33 class ExcludeUnsupportedInt32Test(trt_test.TfTrtIntegrationTestBase):
52 return trt_test.TfTrtIntegrationTestParams(
68 rewriter_config=trt_test.OptimizerDisabledRewriterConfig())
78 return not trt_test.IsQuantizationMode(run_params.precision_mode)
topk_test.py 21 from tensorflow.python.compiler.tensorrt.test import tf_trt_integration_test_base as trt_test
30 class TopKTest(trt_test.TfTrtIntegrationTestBase):
45 return trt_test.TfTrtIntegrationTestParams(
57 class TopKOutputTypeTest(trt_test.TfTrtIntegrationTestBase):
76 return trt_test.TfTrtIntegrationTestParams(
reshape_transpose_test.py 21 from tensorflow.python.compiler.tensorrt.test import tf_trt_integration_test_base as trt_test
29 class ReshapeTest(trt_test.TfTrtIntegrationTestBase):
72 return trt_test.TfTrtIntegrationTestParams(
88 return (not trt_test.IsQuantizationMode(run_params.precision_mode) and
92 class TransposeTest(trt_test.TfTrtIntegrationTestBase):
129 return trt_test.TfTrtIntegrationTestParams(
147 return (not trt_test.IsQuantizationMode(run_params.precision_mode) and
base_test.py 23 from tensorflow.python.compiler.tensorrt.test import tf_trt_integration_test_base as trt_test
34 class SimpleSingleEngineTest(trt_test.TfTrtIntegrationTestBase):
67 return trt_test.TfTrtIntegrationTestParams(
84 class SimpleMultiEnginesTest(trt_test.TfTrtIntegrationTestBase):
124 return trt_test.TfTrtIntegrationTestParams(
147 rewriter_config=trt_test.OptimizerDisabledRewriterConfig())
150 class SimpleMultiEnginesTest2(trt_test.TfTrtIntegrationTestBase):
176 return trt_test.TfTrtIntegrationTestParams(
195 not (trt_test.IsQuantizationMode(run_params.precision_mode) and
199 class ConstInputTest(trt_test.TfTrtIntegrationTestBase)
    [all...]
binary_tensor_weight_broadcast_test.py 23 from tensorflow.python.compiler.tensorrt.test import tf_trt_integration_test_base as trt_test
32 class BinaryTensorWeightBroadcastTest(trt_test.TfTrtIntegrationTestBase):
63 return trt_test.TfTrtIntegrationTestParams(
identity_output_test.py 28 from tensorflow.python.compiler.tensorrt.test import tf_trt_integration_test_base as trt_test
37 class IdentityTest(trt_test.TfTrtIntegrationTestBase):
61 return trt_test.TfTrtIntegrationTestParams(
memory_alignment_test.py 23 from tensorflow.python.compiler.tensorrt.test import tf_trt_integration_test_base as trt_test
32 class MemoryAlignmentTest(trt_test.TfTrtIntegrationTestBase):
62 return trt_test.TfTrtIntegrationTestParams(
neighboring_engine_test.py 23 from tensorflow.python.compiler.tensorrt.test import tf_trt_integration_test_base as trt_test
33 class NeighboringEngineTest(trt_test.TfTrtIntegrationTestBase):
59 return trt_test.TfTrtIntegrationTestParams(
conv2d_test.py 23 from tensorflow.python.compiler.tensorrt.test import tf_trt_integration_test_base as trt_test
91 class Conv2DNCHWTest(trt_test.TfTrtIntegrationTestBase):
104 return trt_test.TfTrtIntegrationTestParams(
116 class Conv2DNHWCTest(trt_test.TfTrtIntegrationTestBase):
129 return trt_test.TfTrtIntegrationTestParams(
141 class Conv2DStridedNCHWTest(trt_test.TfTrtIntegrationTestBase):
179 return trt_test.TfTrtIntegrationTestParams(
191 class Conv2DTranposeTest(trt_test.TfTrtIntegrationTestBase):
220 return trt_test.TfTrtIntegrationTestParams(
lru_cache_test.py 23 from tensorflow.python.compiler.tensorrt.test import tf_trt_integration_test_base as trt_test
33 class LRUCacheTest(trt_test.TfTrtIntegrationTestBase):
61 return trt_test.TfTrtIntegrationTestParams(
74 not trt_test.IsQuantizationMode(run_params.precision_mode))
batch_matmul_test.py 23 from tensorflow.python.compiler.tensorrt.test import tf_trt_integration_test_base as trt_test
33 class BatchMatMulTest(trt_test.TfTrtIntegrationTestBase):
71 return trt_test.TfTrtIntegrationTestParams(
const_broadcast_test.py 23 from tensorflow.python.compiler.tensorrt.test import tf_trt_integration_test_base as trt_test
32 class ConstBroadcastTest(trt_test.TfTrtIntegrationTestBase):
58 return trt_test.TfTrtIntegrationTestParams(
multi_connection_neighbor_engine_test.py 23 from tensorflow.python.compiler.tensorrt.test import tf_trt_integration_test_base as trt_test
32 class MultiConnectionNeighborEngineTest(trt_test.TfTrtIntegrationTestBase):
75 return trt_test.TfTrtIntegrationTestParams(
rank_two_test.py 21 from tensorflow.python.compiler.tensorrt.test import tf_trt_integration_test_base as trt_test
31 class RankTwoTest(trt_test.TfTrtIntegrationTestBase):
63 return trt_test.TfTrtIntegrationTestParams(
vgg_block_nchw_test.py 23 from tensorflow.python.compiler.tensorrt.test import tf_trt_integration_test_base as trt_test
34 class VGGBlockNCHWTest(trt_test.TfTrtIntegrationTestBase):
70 return trt_test.TfTrtIntegrationTestParams(
vgg_block_test.py 23 from tensorflow.python.compiler.tensorrt.test import tf_trt_integration_test_base as trt_test
34 class VGGBlockTest(trt_test.TfTrtIntegrationTestBase):
61 return trt_test.TfTrtIntegrationTestParams(
concatenation_test.py 23 from tensorflow.python.compiler.tensorrt.test import tf_trt_integration_test_base as trt_test
33 class ConcatenationTest(trt_test.TfTrtIntegrationTestBase):
73 return trt_test.TfTrtIntegrationTestParams(
unary_test.py 23 from tensorflow.python.compiler.tensorrt.test import tf_trt_integration_test_base as trt_test
34 class UnaryTest(trt_test.TfTrtIntegrationTestBase):
100 return trt_test.TfTrtIntegrationTestParams(
biasadd_matmul_test.py 23 from tensorflow.python.compiler.tensorrt.test import tf_trt_integration_test_base as trt_test
34 class BiasaddMatMulTest(trt_test.TfTrtIntegrationTestBase):
111 return trt_test.TfTrtIntegrationTestParams(
127 rewriter_config=trt_test.OptimizerDisabledRewriterConfig())

Completed in 742 milliseconds