Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes_test.cc1774 : trt_mode_(std::get<0>(GetParam())), in ParameterizedOpConverterTestBase()
1779 LOG(INFO) << "trt_mode_: " << DebugString(trt_mode_); in ParameterizedOpConverterTestBase()
1785 OpConverterTest::Reset(converter_precision_, trt_mode_); in Reset()
1791 TrtTestMode get_trt_mode() { return trt_mode_; } in get_trt_mode()
1835 if (trt_mode_ == TrtTestMode::kDynamicShape) {
1945 const TrtTestMode trt_mode_; member in tensorflow::tensorrt::convert::ParameterizedOpConverterTestBase
2203 if (trt_mode_ == TrtTestMode::kDynamicShape) { in TEST_P()
2232 (i != 0 && trt_mode_ == TrtTestMode::kImplicitBatch) in TEST_P()
2240 if (i == 0 && trt_mode_ == TrtTestMode::kDynamicShape && in TEST_P()
2293 (trt_mode_ == TrtTestMode::kImplicitBatch) in TEST_P()
[all …]