Home
last modified time | relevance | path

Searched refs:GetReduceAttrAxis (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/format_type/
Dchange_axis_of_reduce_kernel.cc70 auto axis = kernel::GetReduceAttrAxis(cnode); in ConvertReduceAttrFraczAnd6HD()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_select/
Dtbe_kernel_reduce_selecter.cc52 axis_ = GetReduceAttrAxis(cnode_ptr_); in GetShapeInfo()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/
Dcommon_utils.h103 std::vector<int64_t> GetReduceAttrAxis(const CNodePtr &cnode);
Dcommon_utils.cc805 std::vector<int64_t> GetReduceAttrAxis(const CNodePtr &cnode) { in GetReduceAttrAxis() function