Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/
Dadd_atomic_clean.cc64 std::set<int64_t> axis_set(axis_vec.begin(), axis_vec.end()); in GetUniqReduceAxes() local
178 std::set<int64_t> axis_set = GetUniqReduceAxes(node); in SuitableForAtomicAdd() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/math/
Dlp_norm_gpu_kernel.cc111 std::set<size_t> axis_set(axis.begin(), axis.end()); in Resize() local
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/symbol_ops_impl/
Dreduce.cc100 HashSet<int64_t> axis_set; in Eval() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/expanders/
Dutils.cc136 std::unordered_set<int64_t> axis_set(axis.begin(), axis.end()); in GetReducedOriShape() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/
Dlp_norm_cpu_kernel.cc117 std::set<size_t> axis_set(axis.begin(), axis.end()); in Resize() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/core/
Darithmetic_simplify.cc397 std::set<int64_t> axis_set; in UpdateParameters() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/operator/
Dops_front_infer_function.cc482 std::set<int64_t> axis_set; in DoInferReduceShape() local