Searched refs:RenameAndMarkOutputTensors (Results 1 – 3 of 3) sorted by relevance
452 Status RenameAndMarkOutputTensors(
642 TEST_F(ConverterTest, RenameAndMarkOutputTensors) { in TEST_F() argument677 converter_->RenameAndMarkOutputTensors({{"my_op:2", "my_output"}}), in TEST_F()681 TF_EXPECT_OK(converter_->RenameAndMarkOutputTensors( in TEST_F()1124 TF_EXPECT_OK(converter_->RenameAndMarkOutputTensors(output_info)); in BuildAndRun()
1074 Status Converter::RenameAndMarkOutputTensors( in RenameAndMarkOutputTensors() function in tensorflow::tensorrt::convert::Converter4152 TF_RETURN_IF_ERROR(converter.RenameAndMarkOutputTensors(output_tensors)); in ConvertGraphDefToEngine()