Home
last modified time | relevance | path

Searched defs:IntDivider (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/cuda/detail/
H A DIntegerDivider.cuh65 struct IntDivider { struct
67 IntDivider(Value d) : divisor(d) { } in IntDivider() argument
69 C10_HOST_DEVICE inline Value div(Value n) const { return n / divisor; } in div()
70 C10_HOST_DEVICE inline Value mod(Value n) const { return n % divisor; } in mod()
71 C10_HOST_DEVICE inline DivMod<Value> divmod(Value n) const { in divmod()
75 Value divisor;
80 struct IntDivider<unsigned int> { struct
85 IntDivider(unsigned int d) : divisor(d) { in IntDivider() argument
97 C10_HOST_DEVICE inline unsigned int div(unsigned int n) const { in div()
110 C10_HOST_DEVICE inline unsigned int mod(unsigned int n) const { in mod()
[all …]