Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.h425 string source_tensor_name; member
Dconvert_nodes.cc1079 GetTensorOrWeights(output.source_tensor_name, &tensor_or_weights)); in RenameAndMarkOutputTensors()
1081 return errors::InvalidArgument("Output ", output.source_tensor_name, in RenameAndMarkOutputTensors()
1087 output.source_tensor_name); in RenameAndMarkOutputTensors()
1112 VLOG(1) << "Marking output TRT tensor " << output.source_tensor_name in RenameAndMarkOutputTensors()