MCPcopy Create free account
hub / github.com/MegEngine/MegEngine / get_binding_dimensions

Method get_binding_dimensions

src/tensorrt/impl/tensorrt_opr.cpp:285–288  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

283
284#if NV_TENSOR_RT_VERSION >= 6001
285nvinfer1::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
291void TensorRTManager::exec(

Callers 1

get_output_var_shapeMethod · 0.80

Calls

no outgoing calls

Tested by

no test coverage detected