Searched defs:ApplyOp2 (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/cuda/ |
| H A D | CUDAApplyUtils.cuh | 294 struct ApplyOp2 { struct
296 static void apply(detail::TensorInfo<scalar1, IndexType> &a, in apply()
323 struct ApplyOp2<Op, scalar1, scalar2, IndexType, ADims, BDims, 0, Offset> { struct
325 static void apply(detail::TensorInfo<scalar1, IndexType> &a, in apply()
340 struct ApplyOp2<Op, scalar1, scalar2, IndexType, ADims, BDims, 0, Offsets...> { struct
342 static void apply(detail::TensorInfo<scalar1, IndexType> &a, in apply()
|