Home
last modified time | relevance | path

Searched refs:op_converter (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes_test.cc634 auto op_converter = [&start_conversion, in TEST_F() local
647 op_validators(&validator)["Add"] = op_converter; in TEST_F()
747 void AddOpConverter(const string& op_name, OpConverter op_converter) { in AddOpConverter() argument
748 converter_->op_registry_[op_name] = op_converter; in AddOpConverter()
795 auto op_converter = [&output_tensors](OpConverterParams* params) -> Status { in TEST_F() local
813 AddOpConverter("MyOp", op_converter); in TEST_F()
867 auto op_converter = [&output_tensors](OpConverterParams* params) -> Status { in TEST_F() local
884 AddOpConverter("MyOp", op_converter); in TEST_F()
Dconvert_nodes.cc1296 OpConverter op_converter = itr->second; in ConvertNode() local
1297 TF_RETURN_IF_ERROR(op_converter(&params)); in ConvertNode()