| /external/tensorflow/tensorflow/python/compiler/tensorrt/test/ |
| D | base_test.py | 64 def ExpectedEnginesToBuild(self, run_params): member in SimpleSingleEngineTest
111 def ExpectedEnginesToBuild(self, run_params): member in SimpleMultiEnginesTest
151 def ExpectedEnginesToBuild(self, run_params): member in SimpleMultiEnginesTest2
196 def ExpectedEnginesToBuild(self, run_params): member in ConstInputTest
220 def ExpectedEnginesToBuild(self, run_params): member in ConstDataInputSingleEngineTest
245 def ExpectedEnginesToBuild(self, run_params): member in ConstDataInputMultipleEnginesTest
285 def ExpectedEnginesToBuild(self, run_params): member in ControlDependencyTest
|
| D | trt_mode_test.py | 79 def ExpectedEnginesToBuild(self, run_params): member in ImplicitBatchTest
108 def ExpectedEnginesToBuild(self, run_params): member in ExplicitBatchTest
154 def ExpectedEnginesToBuild(self, run_params): member in DynamicShapesTest
|
| D | conv2d_test.py | 101 def ExpectedEnginesToBuild(self, run_params): member in Conv2DNCHWTest
136 def ExpectedEnginesToBuild(self, run_params): member in Conv2DNHWCTest
167 def ExpectedEnginesToBuild(self, run_params): member in Conv2DStridedNCHWTest
197 def ExpectedEnginesToBuild(self, run_params): member in Conv2DTranposeTest
|
| D | batch_matmul_test.py | 70 def ExpectedEnginesToBuild(self, run_params): member in BatchMatMulTwoTensorTest
89 def ExpectedEnginesToBuild(self, run_params): member in BatchMatMulWeightBroadcastTest
107 def ExpectedEnginesToBuild(self, run_params): member in BatchMatMulWeightBroadcastDims2Test
|
| D | reshape_transpose_test.py | 66 def ExpectedEnginesToBuild(self, run_params): member in ReshapeTest
93 def ExpectedEnginesToBuild(self, run_params): member in TransposeTest
122 def ExpectedEnginesToBuild(self, run_params): member in IncompatibleTransposeTest
|
| D | quantization_test.py | 72 def ExpectedEnginesToBuild(self, run_params): member in QuantizationMissingAllRangesTest
96 def ExpectedEnginesToBuild(self, run_params): member in QuantizationWithRangesTest
123 def ExpectedEnginesToBuild(self, run_params): member in NonQuantizedPrecisionsWithRangesTest
|
| D | topk_test.py | 45 def ExpectedEnginesToBuild(self, run_params): member in TopKTest
70 def ExpectedEnginesToBuild(self, run_params): member in TopKOutputTypeTest
|
| D | int32_test.py | 61 def ExpectedEnginesToBuild(self, run_params): member in ExcludeUnsupportedInt32Test
85 def ExpectedEnginesToBuild(self, run_params): member in CalibrationInt32Support
|
| D | identity_output_test.py | 58 def ExpectedEnginesToBuild(self, run_params): member in IdentityTest
|
| D | cast_test.py | 48 def ExpectedEnginesToBuild(self, run_params): member in CastInt32ToFp32Test
|
| D | memory_alignment_test.py | 58 def ExpectedEnginesToBuild(self, run_params): member in MemoryAlignmentTest
|
| D | binary_tensor_weight_broadcast_test.py | 61 def ExpectedEnginesToBuild(self, run_params): member in BinaryTensorWeightBroadcastTest
|
| D | lru_cache_test.py | 60 def ExpectedEnginesToBuild(self, run_params): member in LRUCacheTest
|
| D | neighboring_engine_test.py | 57 def ExpectedEnginesToBuild(self, run_params): member in NeighboringEngineTest
|
| D | const_broadcast_test.py | 51 def ExpectedEnginesToBuild(self, run_params): member in ConstBroadcastTest
|
| D | vgg_block_nchw_test.py | 70 def ExpectedEnginesToBuild(self, run_params): member in VGGBlockNCHWTest
|
| D | vgg_block_test.py | 61 def ExpectedEnginesToBuild(self, run_params): member in VGGBlockTest
|
| D | multi_connection_neighbor_engine_test.py | 71 def ExpectedEnginesToBuild(self, run_params): member in MultiConnectionNeighborEngineTest
|
| D | rank_two_test.py | 62 def ExpectedEnginesToBuild(self, run_params): member in RankTwoTest
|
| D | dynamic_input_shapes_test.py | 89 def ExpectedEnginesToBuild(self, run_params): member in DynamicInputShapesTest
|
| D | annotate_max_batch_sizes_test.py | 68 def ExpectedEnginesToBuild(self, run_params): member in MaxBatchSizesTestBase
|
| D | concatenation_test.py | 71 def ExpectedEnginesToBuild(self, run_params): member in ConcatenationTest
|
| D | unary_test.py | 98 def ExpectedEnginesToBuild(self, run_params): member in UnaryTest
|
| D | combined_nms_test.py | 83 def ExpectedEnginesToBuild(self, run_params): member in CombinedNmsTest
|
| D | biasadd_matmul_test.py | 122 def ExpectedEnginesToBuild(self, run_params): member in BiasaddMatMulTest
|