29 #include <unordered_set>
51 static const std::unordered_set<ReductionOpCode, utility::hash_enum_class>
52 s_regular_reduce_ops = {
58 static const std::unordered_set<ReductionOpCode, utility::hash_enum_class>
63 static const std::unordered_set<ReductionOpCode, utility::hash_enum_class>
64 s_boolean_reduce_ops = {
71 const SizeVector& dims,
77 const SizeVector& dims,
81 #ifdef BUILD_CUDA_MODULE
82 void ReductionCUDA(
const Tensor& src,
84 const SizeVector& dims,
ReductionOpCode
Definition: Reduction.h:40
void ReductionCPU(const Tensor &src, Tensor &dst, const SizeVector &dims, bool keepdim, ReductionOpCode op_code)
Definition: ReductionCPU.cpp:238
void Reduction(const Tensor &src, Tensor &dst, const SizeVector &dims, bool keepdim, ReductionOpCode op_code)
Definition: Reduction.cpp:35
Definition: PinholeCameraIntrinsic.cpp:35