Searched refs:trt_profile_idx (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/ |
| D | trt_engine_utils.cc | 136 const int trt_profile_idx, in SetTrtEngineInputs() argument
146 TF_RETURN_IF_ERROR(GetTrtBindingIndex(input_name.c_str(), trt_profile_idx, in SetTrtEngineInputs()
219 int trt_profile_idx, std::vector<void*>& buffers, in SetTrtEngineOutputs() argument
228 TF_RETURN_IF_ERROR(GetTrtBindingIndex(output_name.c_str(), trt_profile_idx, in SetTrtEngineOutputs()
|
| D | trt_engine_utils.h | 136 const int trt_profile_idx,
156 int trt_profile_idx, std::vector<void*>& buffers,
|