Home
last modified time | relevance | path

Searched refs:x_shape_max (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore/core/abstract/
Dprim_arrays.cc274 ShapeVector x_shape_max; in InferImplUnsortedSegmentSum() local
276 x_shape_max = (x_is_dyn) ? x->shape()->max_shape() : x->shape()->shape(); in InferImplUnsortedSegmentSum()
278 …max_shape.insert(max_shape.end(), x_shape_max.begin() + segment_ids_shape.size(), x_shape_max.end(… in InferImplUnsortedSegmentSum()
325 ShapeVector x_shape_max; in InferImplUnsortedSegmentMax() local
327 x_shape_max = (x_is_dyn) ? x->shape()->max_shape() : x->shape()->shape(); in InferImplUnsortedSegmentMax()
329 …max_shape.insert(max_shape.end(), x_shape_max.begin() + segment_ids_shape.size(), x_shape_max.end(… in InferImplUnsortedSegmentMax()
377 ShapeVector x_shape_max; in InferImplUnsortedSegmentMin() local
379 x_shape_max = (x_is_dyn) ? x->shape()->max_shape() : x->shape()->shape(); in InferImplUnsortedSegmentMin()
381 …max_shape.insert(max_shape.end(), x_shape_max.begin() + segment_ids_shape.size(), x_shape_max.end(… in InferImplUnsortedSegmentMin()
837 ShapeVector x_shape_max = input_x->shape()->max_shape(); in InferImplSplit() local
[all …]
Dprim_maths.cc154 ShapeVector x_shape_max = input_x->shape()->max_shape(); in InferImplReduceFunc() local
156 InferImplReduceFuncCalShape(&shape_max, x_shape_max, axis, keep_dims_value); in InferImplReduceFunc()
422 ShapeVector x_shape_max = x->shape()->max_shape().empty() ? x_shape : x->shape()->max_shape(); in InferImplLess() local
433 auto out_shape_max = BroadcastShape(x_shape_max, y_shape_max); in InferImplLess()