Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/cuda/
H A DSparseCsrTensorMath.cu700 struct ReductionMulOp { struct
701 __device__ __forceinline__ scalar_t operator()(const scalar_t a, const scalar_t b) const { in operator ()()
704 __device__ __forceinline__ scalar_t identity() const { return 1; } in identity()
705 __forceinline__ scalar_t identity_cpu() const { return 1; } in identity_cpu()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/
H A DSparseCsrTensorMath.cpp1322 struct ReductionMulOp { struct
1323 inline scalar_t operator()(const scalar_t& a, const scalar_t& b) const { in operator ()()
1326 inline scalar_t identity() const { return 1; } in identity()