Mojo function
get_cudnn_dtype
get_cudnn_dtype[dtype: DType]() -> cudnnDataType_t
Map Mojo DType to cuDNN data type.
Support only floating point dtypes for now.
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!