Searched defs:ReduceAll (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/ |
| D | reduce_impl.cuh | 30 ReduceAll = 6, enumerator
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/gpu/ |
| D | test_reduce_all_op.py | 43 class ReduceAll(nn.Cell): class
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32/ |
| D | reduce_fp32.c | 162 int ReduceAll(int outer_size, int inner_size, int axis_size, const bool *src_data, bool *dst_data, … in ReduceAll() function
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | reduce_cpu_kernel.cc | 223 void ReduceAll(const T *in, T *out, size_t start, size_t end, TransposeIterator *iter) { in ReduceAll() function
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/auto_generate/ |
| D | gen_lite_ops.h | 1933 ReduceAll() : BaseOperator(kNameReduceAll) {} in ReduceAll() function
|