Searched refs:TRTEngineOpTestBase (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| D | trt_engine_op_test.cc | 60 class TRTEngineOpTestBase : public OpsTestBase { class
153 TEST_F(TRTEngineOpTestBase, DynamicEngines) { in TEST_F() argument
155 TRTEngineOpTestBase::AddSimpleTrtOp(DT_FLOAT, /*max_cached_engines_count=*/4); in TEST_F()
158 TRTEngineOpTestBase::AddSimpleInput<float>(TensorShape({2, 2})); in TEST_F()
175 TRTEngineOpTestBase::AddSimpleInput<float>(TensorShape({1, 2})); in TEST_F()
182 TRTEngineOpTestBase::AddSimpleInput<float>(TensorShape({3, 2})); in TEST_F()
190 TRTEngineOpTestBase::AddSimpleInput<float>(TensorShape({10, 10})); in TEST_F()
195 TRTEngineOpTestBase::AddSimpleInput<float>(TensorShape({1, 10})); in TEST_F()
203 TEST_F(TRTEngineOpTestBase, AllowBuildAtRuntime) { in TEST_F() argument
204 TRTEngineOpTestBase::AddSimpleTrtOp(DT_FLOAT, /*max_cached_engines_count=*/1, in TEST_F()
[all …]
|