Function cudarc::cudnn::result::set_tensor4d_descriptor
source · pub unsafe fn set_tensor4d_descriptor(
tensor_desc: cudnnTensorDescriptor_t,
format: cudnnTensorFormat_t,
data_type: cudnnDataType_t,
[n, c, h, w]: [c_int; 4]
) -> Result<(), CudnnError>
Expand description
Sets data on a tensor descriptor. See nvidia docs
Safety
tensor_desc
must have been created with create_tensor_descriptor, and
NOT freed by destroy_tensor_descriptor