Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.h718 const std::unordered_map<string, nvinfer1::ActivationType>* ActivationTypeMap();
Dconvert_nodes.cc3561 auto op_pair = ActivationTypeMap()->find(activation_mode); in ConvertFusedConv2DBiasActivation()
3562 if (op_pair == ActivationTypeMap()->end() && activation_mode != "None") { in ConvertFusedConv2DBiasActivation()
3655 if (op_pair != ActivationTypeMap()->end()) { in ConvertFusedConv2DBiasActivation()
3885 ActivationTypeMap() { in ActivationTypeMap() function
3907 auto op_pair = ActivationTypeMap()->find(node_def.op()); in ConvertActivation()
3908 if (op_pair == ActivationTypeMap()->end()) { in ConvertActivation()
6282 for (const auto& activation_op_pair : *ActivationTypeMap()) { in RegisterValidatableOpConverters()
Dconvert_nodes_test.cc3371 auto* map = ActivationTypeMap(); in TEST_P()