Searched defs:ApplyOp1 (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/cuda/ |
| H A D | CUDAApplyUtils.cuh | 225 struct ApplyOp1 { struct
227 static void apply(detail::TensorInfo<scalar, IndexType> &a, const Op &op, int n, in apply()
246 struct ApplyOp1<Op, scalar, IndexType, ADims, 0, Offset> { struct
248 static void apply(detail::TensorInfo<scalar, IndexType> &a, const Op &op, in apply()
259 struct ApplyOp1<Op, scalar, IndexType, ADims, 0, Offsets...> { struct
261 static void apply(detail::TensorInfo<scalar, IndexType> &a, const Op &op, int n, in apply()
|