10#include <unordered_set>
32static const std::unordered_set<ReductionOpCode, utility::hash_enum_class>
33 s_regular_reduce_ops = {
39static const std::unordered_set<ReductionOpCode, utility::hash_enum_class>
44static const std::unordered_set<ReductionOpCode, utility::hash_enum_class>
45 s_boolean_reduce_ops = {
52 const SizeVector& dims,
58 const SizeVector& dims,
62#ifdef BUILD_SYCL_MODULE
65 const SizeVector& dims,
70#ifdef BUILD_CUDA_MODULE
71void ReductionCUDA(
const Tensor& src,
73 const SizeVector& dims,
ReductionOpCode
Definition Reduction.h:21
void ReductionSYCL(const Tensor &src, Tensor &dst, const SizeVector &dims, bool keepdim, ReductionOpCode op_code)
Definition ReductionSYCL.cpp:210
void ReductionCPU(const Tensor &src, Tensor &dst, const SizeVector &dims, bool keepdim, ReductionOpCode op_code)
Definition ReductionCPU.cpp:277
void Reduction(const Tensor &src, Tensor &dst, const SizeVector &dims, bool keepdim, ReductionOpCode op_code)
Definition Reduction.cpp:16
Definition PinholeCameraIntrinsic.cpp:16