Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dutils.h42 static constexpr const char* const kOutputPHName = "TensorRTOutputPH_"; variable
Dconvert_nodes.cc170 return absl::StartsWith(name, IONamePrefixes::kOutputPHName); in IsEngineOutput()
6392 node_name.c_str() + strlen(IONamePrefixes::kOutputPHName), in ConvertGraphDefToEngine()
6511 StrCat(IONamePrefixes::kOutputPHName, connection.port_number); in ConvertSegmentToGraphDef()
Dconvert_nodes_test.cc1311 output = ops::Identity(s.WithOpName(StrCat(IONamePrefixes::kOutputPHName, 0)), in TEST_F()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/
Dtrt_engine_utils.cc226 ctx ? StrCat(IONamePrefixes::kOutputPHName, i) : outputs->at(i).name; in SetTrtEngineOutputs()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/
Dtrt_engine_op.cc257 if (replace_name(IONamePrefixes::kOutputPHName, node.mutable_name())) { in FunctionDefToGraphDef()
269 replace_name(IONamePrefixes::kOutputPHName, &input); in FunctionDefToGraphDef()