Searched refs:in1_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
127 auto in1_iter = in1_accum.begin(); in UnifyAccumulateProduct()
129 while ((in1_iter < in1_accum.end()) || (in2_iter < in2_accum.end())) { in UnifyAccumulateProduct()
147 if ((in1_iter != in1_accum.end()) || (in2_iter != in2_accum.end())) { in UnifyAccumulateProduct()
162 Shape in1_accum; in UnifyShape() local
163 Status status = ShapeToAccumulateProduct(in1, &in1_accum); in UnifyShape()
173 status = UnifyAccumulateProduct(in1_accum, in2_accum, &out_accum); in UnifyShape()
|
| D | arrangement.cc | 194 std::vector<int64_t> in1_accum; in GetUnifiedShape() local
195 Status status = ShapeToAccumulateProduct(array_, &in1_accum); in GetUnifiedShape()
205 status = UnifyAccumulateProduct(in1_accum, in2_accum, &out_accum); in GetUnifiedShape()
|
| D | shape_util.h | 105 Status UnifyAccumulateProduct(const Shape &in1_accum, const Shape &in2_accum, Shape *out_accum);
|