Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/core/ops/
Dreduce_sum.cc96 ShapeVector out_min_shape = {}; in InferShape() local
112 out_min_shape.push_back(1); in InferShape()
118 out_min_shape.push_back(1); in InferShape()
124 out_min_shape.push_back(1); in InferShape()
128 return std::make_shared<abstract::Shape>(out_shape, out_min_shape, out_max_shape); in InferShape()
Ddropout_gen_mask.cc133 ShapeVector out_min_shape = CalDynamicOutputShape(min_value); in InferShape() local
137 return std::make_shared<abstract::Shape>(any_shape, out_min_shape, out_max_shape); in InferShape()