Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.h42 extern const char* const kOutputPHName;
Dconvert_graph.cc643 } else if (str_util::StartsWith(n->name(), kOutputPHName)) { in RegisterSegmentFunctionToFunctionLibrary()
676 auto name = StrCat(kOutputPHName, i); in RegisterSegmentFunctionToFunctionLibrary()
Dconvert_nodes.cc84 const char* const kOutputPHName = "TensorRTOutputPH_"; variable
90 return absl::StartsWith(name, kOutputPHName); in IsEngineOutput()
4132 node_name.c_str() + strlen(kOutputPHName), &slot_number)) { in ConvertGraphDefToEngine()
4222 const string node_name = StrCat(kOutputPHName, connection.port_number); in ConvertSegmentToGraphDef()
Dconvert_nodes_test.cc1031 output = ops::Identity(s.WithOpName(StrCat(kOutputPHName, 0)), output); in TEST_F()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/
Dtrt_engine_op.cc446 const string output_name = StrCat(kOutputPHName, i); in ExecuteTrtEngine()