Home
last modified time | relevance | path

Searched defs:reduce_sum_shape (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fission/
Ddynamic_rnn_grad_fission_v2.cc751 auto reduce_sum_shape = in CreateDwReduceSum() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/auto_parallel/
Doperator_costmodel.cc1977 Shape reduce_sum_shape = {box_shape[0], crop_size_[0], crop_size_[1], x_shape[3]}; in GetForwardCommCost() local
2085 Shape reduce_sum_shape = {rois_shape[0], features_shape[1], pooled_shape_[0], pooled_shape_[1]}; in GetForwardCommCost() local