Function cudarc::cudnn::result::set_tensornd_descriptor
source · pub unsafe fn set_tensornd_descriptor(
tensor_desc: cudnnTensorDescriptor_t,
data_type: cudnnDataType_t,
num_dims: c_int,
dims: *const c_int,
strides: *const c_int
) -> 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