Home
last modified time | relevance | path

Searched defs:GetReduceAxis (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/expanders/
Dminimum_grad.cc31 …static ShapeVector GetReduceAxis(const ShapeVector &original_shape, const ShapeVector &brodcast_sh… in GetReduceAxis() function in mindspore::graphkernel::expanders::MinimumGrad
Dmaximum_grad.cc31 …static ShapeVector GetReduceAxis(const ShapeVector &original_shape, const ShapeVector &brodcast_sh… in GetReduceAxis() function in mindspore::graphkernel::expanders::MaximumGrad
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/
Dgraph_kernel_helper.cc231 std::vector<int64_t> GetReduceAxis(const AnfNodePtr &node) { in GetReduceAxis() function