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.cc485 auto op_converter = [&start_conversion, in TEST_F() local
499 op_validators()["MyOp"] = op_converter; in TEST_F()
529 void AddOpConverter(const string& op_name, OpConverter op_converter) { in AddOpConverter() argument
530 converter_->op_registry_[op_name] = op_converter; in AddOpConverter()
581 auto op_converter = [&output_tensors](OpConverterParams* params) -> Status { in TEST_F() local
599 AddOpConverter("MyOp", op_converter); in TEST_F()
649 auto op_converter = [&output_tensors](OpConverterParams* params) -> Status { in TEST_F() local
667 AddOpConverter("MyOp", op_converter); in TEST_F()
Dconvert_nodes.cc1014 OpConverter op_converter = op_registry_.at(op); in ConvertNode() local
1015 TF_RETURN_IF_ERROR(op_converter(&params)); in ConvertNode()