Searched refs:out_accum (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/tensor_layout/ |
| D | shape_util.cc | 124 Status UnifyAccumulateProduct(const Shape &in1_accum, const Shape &in2_accum, Shape *out_accum) { in UnifyAccumulateProduct() argument
125 MS_EXCEPTION_IF_NULL(out_accum); in UnifyAccumulateProduct()
126 out_accum->clear(); in UnifyAccumulateProduct()
135 out_accum->push_back(*in1_iter); in UnifyAccumulateProduct()
139 out_accum->push_back(*in1_iter); in UnifyAccumulateProduct()
143 out_accum->push_back(*in2_iter); in UnifyAccumulateProduct()
172 Shape out_accum; in UnifyShape() local
173 status = UnifyAccumulateProduct(in1_accum, in2_accum, &out_accum); in UnifyShape()
177 status = AccumulateProductToShape(out_accum, out); in UnifyShape()
|
| D | arrangement.cc | 204 std::vector<int64_t> out_accum; in GetUnifiedShape() local
205 status = UnifyAccumulateProduct(in1_accum, in2_accum, &out_accum); in GetUnifiedShape()
210 status = AccumulateProductToShape(out_accum, &out_shape); in GetUnifiedShape()
|
| D | shape_util.h | 105 Status UnifyAccumulateProduct(const Shape &in1_accum, const Shape &in2_accum, Shape *out_accum);
|