Function SetTensorFromVector

tests/common/TensorDataUtils.hpp:490–524

template<typename DT>
void SetTensorFromVector(const TensorData &tensorData, std::vector<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 ((int64_t)(data.size() * sizeof(DT)) != tDataAc->sampleStride())
        throw std::runtime_error("Data vector is incorrect size.");

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

    if (sample < 0)
    {
        for (int i = 0; i < tDataAc->numSamples(); ++i)
        {
            if (cudaSuccess
                != cudaMemcpy(tDataAc->sampleData(i), data.data(), tDataAc->sampleStride(), cudaMemcpyHostToDevice))
            {
                throw std::runtime_error("CudaMemcpy failed");
            }
        }
    }
    else
    {
        if (cudaSuccess
            != cudaMemcpy(tDataAc->sampleData(sample), data.data(), tDataAc->sampleStride(), cudaMemcpyHostToDevice))
        {
            throw std::runtime_error("CudaMemcpy failed");
        }
    }

    return;
}

template<typename DT>
void GetVectorFromTensor(const TensorData &tensorData, int sample, std::vector<DT> &outData)