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 = {
81 #ifdef BUILD_CUDA_MODULE 82 void ReductionCUDA(
const Tensor& src,
Definition: SizeVector.h:102
void ReductionCPU(const Tensor &src, Tensor &dst, const SizeVector &dims, bool keepdim, ReductionOpCode op_code)
Definition: ReductionCPU.cpp:235
void Reduction(const Tensor &src, Tensor &dst, const SizeVector &dims, bool keepdim, ReductionOpCode op_code)
Definition: Reduction.cpp:35
Definition: PinholeCameraIntrinsic.cpp:35
ReductionOpCode
Definition: Reduction.h:40