Searched refs:kReduceSum (Results 1 – 3 of 3) sorted by relevance
77 reduce_type_ = kReduceSum; in InitKernel()89 …if ((reduce_type_ == kReduceMean || reduce_type_ == kReduceSum) && axis_.size() == 1 && axis_[0] =… in InitKernel()
41 enum ReduceType { kReduceAll, kReduceAny, kReduceMax, kReduceMin, kReduceSum, kReduceMean }; enumerator
46 const auto kReduceSum = prim::kPrimReduceSum->name(); in GetDependsFormMap() local73 (void)dynamic_shape_depends.emplace(kReduceSum, std::vector<int64_t>{1}); in GetDependsFormMap()