Home
last modified time | relevance | path

Searched defs:reduce_axis (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/tests/st/graph_kernel/model/
Dtest_graph_parallel.py27 def reduce_graph(shape, reduce_axis): argument
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/
Dmean_infer.c54 bool reduce_axis = false; in MeanInferShape() local
Dreduce_infer.c34 bool reduce_axis = false; in ReduceOnSelectedAxes() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/format_type/
Dchange_axis_of_reduce_kernel.cc40 void SafeCheckFunction(const CNodePtr &cnode, const std::vector<int64_t> &reduce_axis) { in SafeCheckFunction()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Dreduce_fp32.c437 bool reduce_axis = false; in ReduceInferShape() local