Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc166 bool IsEngineInput(absl::string_view name) { in IsEngineInput() function
1310 if (!IsEngineInput(tensor_name)) { in ConvertNode()
1385 if (IsEngineInput(tensor->getName()) || IsEngineOutput(tensor->getName())) { in RenameAndMarkOutputTensors()
6345 if (IsEngineInput(node_name)) { in ConvertGraphDefToEngine()
6572 !IsEngineInput(input.first)) { in ConvertSegmentToGraphDef()