Home
last modified time | relevance | path

Searched refs:IntArrayRefCaster (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/transformers/hip/
H A Daotriton_adapter.h57 struct IntArrayRefCaster { struct
62 struct IntArrayRefCaster<TargetType, 1> {
69 struct IntArrayRefCaster<TargetType, 2> {
79 struct IntArrayRefCaster<TargetType, 3> {
90 struct IntArrayRefCaster<TargetType, 4> {
113 IntArrayRefCaster<uint64_t, Rank>::cast(q.sizes()),
114 IntArrayRefCaster<uint64_t, Rank>::cast(strides),