Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.h586 const std::unordered_map<string, nvinfer1::UnaryOperation>* UnaryOperationMap();
Dconvert_nodes.cc3233 UnaryOperationMap() { in UnaryOperationMap() function
3267 auto op_pair = UnaryOperationMap()->find(node_def.op()); in ConvertUnary()
3268 if (op_pair == UnaryOperationMap()->end()) { in ConvertUnary()
4041 for (auto unary_op_pair : *UnaryOperationMap()) { in RegisterValidatableOpConverters()
Dconvert_nodes_test.cc3689 auto* map = UnaryOperationMap(); in TEST_F()