Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes_test.cc86 enum class TrtTestMode { enum
92 string DebugString(const TrtTestMode mode) { in DebugString()
94 case TrtTestMode::kImplicitBatch: in DebugString()
96 case TrtTestMode::kExplicitBatch: in DebugString()
98 case TrtTestMode::kDynamicShape: in DebugString()
113 constexpr std::array<TrtTestMode, 3> ValidTrtModes = {
114 TrtTestMode::kImplicitBatch, TrtTestMode::kExplicitBatch,
115 TrtTestMode::kDynamicShape};
117 constexpr std::array<TrtTestMode, 1> ValidTrtModes = {
118 TrtTestMode::kImplicitBatch};
[all …]