grain ~master (2021-05-15T05:53:16Z)

makeCudnnTensor

grain cudnn

convert contiguous cuda storage to 1-D tensor disc

  1. auto makeCudnnTensor(Variable!(T, dim, DeviceStorage) x)
  2. auto makeCudnnTensor(T storage)
    version(grain_cuda)
    auto
    makeCudnnTensor
    (
    T
    )
    (
    ref T storage
    )
    if (
    isDeviceMemory!T
    )

Meta

Source

See Implementation

grain cudnn functionsstatic variablesstructs

Page generated by adrdox