Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.h367 using OpConverter = std::function<Status(OpConverterParams*)>; variable
409 std::unordered_map<string, OpConverter> op_validators_;
541 std::unordered_map<string, OpConverter> op_registry_;
547 OpConverter plugin_converter_;
Dconvert_nodes.cc973 OpConverter validator = op_validators_[node_def.op()]; in ValidateNode()
1014 OpConverter op_converter = op_registry_.at(op); in ConvertNode()
3989 std::unordered_map<string, OpConverter>* registration) { in RegisterValidatableOpConverters()
Dconvert_nodes_test.cc397 std::unordered_map<string, OpConverter>& op_validators() { in op_validators()
529 void AddOpConverter(const string& op_name, OpConverter op_converter) { in AddOpConverter()