Function SetTensorTo

tests/common/TensorDataUtils.hpp:359–397

template<typename DT>
void SetTensorTo(const TensorData &tensorData, DT data, int sample)
{
    if (!nvcv::TensorDataAccessStrided::IsCompatible(tensorData))
        throw std::runtime_error("Tensor Data is not pitch access capable.");

    auto tDataAc = nvcv::TensorDataAccessStrided::Create(tensorData);

    if (tDataAc->numSamples() <= sample)
        throw std::runtime_error("Number of samples smaller than requested sample.");

    int             inElements = (tDataAc->sampleStride() / sizeof(DT));
    std::vector<DT> srcVec(inElements, data);

    int totalSamples;
    if (sample < 0)
    {
        totalSamples = tDataAc->numSamples();
        sample       = 0;
    }
    else
    {
        totalSamples = sample + 1;
    }

    for (int i = sample; i < totalSamples; ++i)
    {
        auto  *outSamplePtr = tDataAc->sampleData(i);
        size_t size         = tDataAc->sampleStride();
        if (auto err = cudaMemcpy(outSamplePtr, srcVec.data(), size, cudaMemcpyHostToDevice))
        {
            char msg[1024] = {};
            snprintf(msg, sizeof(msg), "CudaMemcpy failed with %s (%i): %s", cudaGetErrorName(err), err,
                     cudaGetErrorString(err));
            throw std::runtime_error(msg);
        }
    }

    return;
}

template<typename DT>
static void SetTensorToRandomValueFloat(const TensorData &tensorData, DT minVal, DT maxVal, int sample)