Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/cuda/
H A DSparseCsrTensorMath.cu691 struct ReductionAddOp { struct
692 __device__ __forceinline__ scalar_t operator()(const scalar_t a, const scalar_t b) const { in operator ()()
695 __device__ __forceinline__ scalar_t identity() const { return 0; } in identity()
696 __forceinline__ scalar_t identity_cpu() const { return 0; } in identity_cpu()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/
H A DSparseCsrTensorMath.cpp1314 struct ReductionAddOp { struct
1315 inline scalar_t operator()(const scalar_t& a, const scalar_t& b) const { in operator ()()
1318 inline scalar_t identity() const { return 0; } in identity()