Searched refs:source_tensor_name (Results 1 – 2 of 2) sorted by relevance
446 string source_tensor_name; member
1365 GetTensorOrWeights(output.source_tensor_name, &tensor_or_weights)); in RenameAndMarkOutputTensors()1367 return errors::InvalidArgument("Output ", output.source_tensor_name, in RenameAndMarkOutputTensors()1373 output.source_tensor_name); in RenameAndMarkOutputTensors()1400 VLOG(1) << "Marking output TRT tensor " << output.source_tensor_name in RenameAndMarkOutputTensors()