| 283 | |
| 284 | #if NV_TENSOR_RT_VERSION >= 6001 |
| 285 | nvinfer1::Dims TensorRTManager::get_binding_dimensions(int binding_idx) const { |
| 286 | mgb_assert(m_context, "Please create_trt_context before get_binding_dimensions."); |
| 287 | return m_context->getBindingDimensions(binding_idx + m_offset); |
| 288 | } |
| 289 | #endif |
| 290 | |
| 291 | void TensorRTManager::exec( |
no outgoing calls
no test coverage detected