MCPcopy Create free account
hub / github.com/arrayfire/arrayfire / toCudnn

Function toCudnn

src/backend/cuda/convolveNN.cpp:54–58  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

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

Callers

nothing calls this directly

Calls 2

cudnnSetFunction · 0.85
dimsMethod · 0.45

Tested by

no test coverage detected