Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/
Dtrt_engine_utils.cc105 Status GetTrtBindingIndex(const char* tensor_name, int profile_index, in GetTrtBindingIndex() function
146 TF_RETURN_IF_ERROR(GetTrtBindingIndex(input_name.c_str(), trt_profile_idx, in SetTrtEngineInputs()
228 TF_RETURN_IF_ERROR(GetTrtBindingIndex(output_name.c_str(), trt_profile_idx, in SetTrtEngineOutputs()
Dtrt_engine_utils.h128 Status GetTrtBindingIndex(const char* tensor_name, int profile_index,