Searched refs:RenameAndMarkOutputTensors (Results 1 – 3 of 3) sorted by relevance
475 Status RenameAndMarkOutputTensors(
860 TEST_F(ConverterTest, RenameAndMarkOutputTensors) { in TEST_F() argument894 converter_->RenameAndMarkOutputTensors({{"my_op:2", "my_output"}}), in TEST_F()898 TF_EXPECT_OK(converter_->RenameAndMarkOutputTensors( in TEST_F()1472 TF_RETURN_IF_ERROR(converter_->RenameAndMarkOutputTensors(output_info)); in BuildAndRun()
1359 Status Converter::RenameAndMarkOutputTensors( in RenameAndMarkOutputTensors() function in tensorflow::tensorrt::convert::Converter6439 TF_RETURN_IF_ERROR(converter->RenameAndMarkOutputTensors(output_tensors)); in ConvertGraphDefToEngine()