Home
last modified time | relevance | path

Searched refs:ReduceSumInfer (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore/core/ops/
Dreduce_sum.h42 AbstractBasePtr ReduceSumInfer(const abstract::AnalysisEnginePtr &, const PrimitivePtr &primitive,
Dreduce_sum.cc186 AbstractBasePtr ReduceSumInfer(const abstract::AnalysisEnginePtr &, const PrimitivePtr &primitive, in ReduceSumInfer() function
/third_party/mindspore/mindspore/core/abstract/
Dprimitive_infer_map.cc221 {prim::kPrimReduceSum, {ops::ReduceSumInfer, nullptr, true}}, in GetPrimitiveToBackendEvalImplMap()