Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/
Dadd_atomic_clean.cc63 std::set<int64_t> axis_set(axis_vec.begin(), axis_vec.end()); in GetUniqReduceAxes() local
173 std::set<int64_t> axis_set = GetUniqReduceAxes(node); in SuitableForAtomicAdd() local
Darithmetic_simplify.cc376 std::set<int64_t> axis_set; in SetAttributes() local
/third_party/mindspore/mindspore/ccsrc/frontend/operator/
Dops_front_infer_function.cc218 std::set<int64_t> axis_set; in DoInferReduceShape() local