Method getTensorInfo

samples/common/sampleInference.cpp:166–192

template <>
void FillBindingClosure<nvinfer1::ICudaEngine, nvinfer1::IExecutionContext>::getTensorInfo(TensorInfo& tensorInfo)
{
    auto const b = tensorInfo.bindingIndex;
    auto const name = engine->getBindingName(b);
    tensorInfo.name = name;
    if (engine->hasImplicitBatchDimension())
    {
        tensorInfo.dims = context->getBindingDimensions(b);
        tensorInfo.comps = engine->getBindingComponentsPerElement(b);
        tensorInfo.strides = context->getStrides(b);
        tensorInfo.vectorDimIndex = engine->getBindingVectorizedDim(b);
        tensorInfo.isInput = engine->bindingIsInput(b);
        tensorInfo.dataType = engine->getBindingDataType(b);
    }
    else
    {
        // Use enqueueV3.
        tensorInfo.dims = context->getTensorShape(name);
        tensorInfo.isDynamic = std::any_of(
            tensorInfo.dims.d, tensorInfo.dims.d + tensorInfo.dims.nbDims, [](int32_t dim) { return dim == -1; });
        tensorInfo.comps = engine->getTensorComponentsPerElement(name);
        tensorInfo.strides = context->getTensorStrides(name);
        tensorInfo.vectorDimIndex = engine->getTensorVectorizedDim(name);
        tensorInfo.isInput = engine->getTensorIOMode(name) == TensorIOMode::kINPUT;
        tensorInfo.dataType = engine->getTensorDataType(name);
    }
}

template <>
void FillBindingClosure<nvinfer1::safe::ICudaEngine, nvinfer1::safe::IExecutionContext>::getTensorInfo(