OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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