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.h384 using OpConverter = std::function<Status(OpConverterParams*)>; variable
419 std::unordered_map<string, OpConverter> op_validators_;
635 std::unordered_map<string, OpConverter> op_registry_;
Dconvert_nodes.cc1221 OpConverter validator = op_validators_[op]; in IsTensorRTCandidate()
1296 OpConverter op_converter = itr->second; in ConvertNode()
6224 std::unordered_map<string, OpConverter>* registration) { in RegisterValidatableOpConverters()
Dconvert_nodes_test.cc529 std::unordered_map<string, OpConverter>& op_validators( in op_validators()
747 void AddOpConverter(const string& op_name, OpConverter op_converter) { in AddOpConverter()