Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.h541 std::unordered_map<string, OpConverter> op_registry_; variable
Dconvert_nodes.cc1011 if (!op_registry_.count(op)) { in ConvertNode()
1014 OpConverter op_converter = op_registry_.at(op); in ConvertNode()
4051 RegisterValidatableOpConverters(&op_registry_); in RegisterOpConverters()
Dconvert_nodes_test.cc530 converter_->op_registry_[op_name] = op_converter; in AddOpConverter()