Searched defs:reduce_dim (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/tensor_layout/ | ||
| D | tensor_info.h | 48 std::vector<int64_t> reduce_dim() const { return reduce_dim_; } in reduce_dim() function |
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/ | ||
| D | reduce_method_info.cc | 43 std::vector<int64_t> ReduceMethod::reduce_dim() { in reduce_dim() function in mindspore::parallel::ReduceMethod |