Compute the offset for the given coordinates in a cube with a depth of 1.
More...
#include <shape.h>
|
static CUTLASS_DEVICE int | get (int d, int h, int w, int c) |
|
template<int kSh_, int kSw_, int kSc_>
struct cutlass::ComputeOffsetFromShape< Shape< 1, kSh_, kSw_, kSc_ > >
- Template Parameters
-
kSh | Elements in the H dimension |
kSw | Elements in the W dimension |
kSc | Separation between two elements in "elements" |
◆ get()
template<int kSh_, int kSw_, int kSc_>
The documentation for this struct was generated from the following file: