Searched refs:w_min_shape (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/core/ops/ |
| D | conv2d.cc | 161 auto w_min_shape = w_shape_map[kMinShape]; in Conv2dInferShape() local
164 CheckAndConvertUtils::CheckMinMaxShape(w_shape, &w_min_shape, &w_max_shape); in Conv2dInferShape()
169 CheckShapeAllPositive(prim_name + " w_min_shape", w_min_shape); in Conv2dInferShape()
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | prim_nn.cc | 235 ShapeVector w_min_shape = input_w->shape()->min_shape(); in CheckShape() local
237 CheckMinMaxShape(w_shape, &w_min_shape, &w_max_shape); in CheckShape()
239 CheckShapeAllPositive(op_name + " w_min_shape", w_min_shape); in CheckShape()
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | _quant_ops.py | 1703 def infer_shape(self, w_shape, w_min_shape, w_max_shape): argument
1704 validator.check_int(len(w_min_shape), len(w_shape), Rel.EQ, "dims of w_min", self.name)
|