Searched refs:kInputPHName (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| D | trt_engine_op.cc | 319 input_data.emplace(StrCat(kInputPHName, i), data_address); in ExecuteCalibration()
405 const string input_name = StrCat(kInputPHName, i); in ExecuteTrtEngine()
679 StrCat(kInputPHName, i), in AllocateCalibrationResources()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.h | 41 extern const char* const kInputPHName;
|
| D | convert_graph.cc | 640 if (str_util::StartsWith(n->name(), kInputPHName)) { in RegisterSegmentFunctionToFunctionLibrary()
649 auto name = StrCat(kInputPHName, i); in RegisterSegmentFunctionToFunctionLibrary()
|
| D | convert_nodes.cc | 83 const char* const kInputPHName = "TensorRTInputPH_"; variable
87 return absl::StartsWith(name, kInputPHName); in IsEngineInput()
4104 node_name.c_str() + strlen(kInputPHName), &slot_number)) { in ConvertGraphDefToEngine()
4203 const string node_name = StrCat(kInputPHName, connection.port_number); in ConvertSegmentToGraphDef()
4261 StrCat(kInputPHName, connection.port_number); in ConvertSegmentToGraphDef()
|
| D | convert_nodes_test.cc | 997 if (str_util::ConsumePrefix(&node_name, kInputPHName)) { in RunConvertGraphDefToEngine()
1027 auto input = ops::Placeholder(s.WithOpName(StrCat(kInputPHName, 0)), DT_FLOAT, in TEST_F()
|