| 52 | |
| 53 | template<typename Desc, typename T> |
| 54 | auto toCudnn(Array<T> arr) { |
| 55 | auto descriptor = make_handle<Desc>(); |
| 56 | cudnnSet(descriptor, getCudnnDataType<T>(), arr.dims()); |
| 57 | return descriptor; |
| 58 | } |
| 59 | |
| 60 | template<typename T> |
| 61 | using scale_type = |