Searched refs:h_axis (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore/core/ops/ |
| D | conv2d.cc | 173 uint64_t h_axis = 2; in Conv2dInferShape() local
178 h_axis = 1; in Conv2dInferShape()
197 if ((w_shape[h_axis] != Shape::SHP_ANY) && (w_shape[h_axis] != kernel_size[0])) { in Conv2dInferShape()
198 …MS_LOG(EXCEPTION) << "weight height = " << w_shape[h_axis] << ", must equal to = " << kernel_size[… in Conv2dInferShape()
214 …Conv2DPadFunction(&output_hw, &pad_list, x_shape[h_axis], x_shape[w_axis], kernel_size, stride, di… in Conv2dInferShape()
216 …Conv2DPadFunction(&output_hw_min, &pad_list_min, x_min_shape[h_axis], x_min_shape[w_axis], kernel_… in Conv2dInferShape()
218 …Conv2DPadFunction(&output_hw_max, &pad_list_max, x_max_shape[h_axis], x_max_shape[w_axis], kernel_… in Conv2dInferShape()
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | prim_nn.cc | 265 uint64_t h_axis = 2; in InferImplConv2D() local
270 h_axis = 1; in InferImplConv2D()
286 if ((w_shape[h_axis] != Shape::SHP_ANY) && (w_shape[h_axis] != kernel_size[0])) { in InferImplConv2D()
287 …MS_LOG(EXCEPTION) << "weight height = " << w_shape[h_axis] << ", must equal to = " << kernel_size[… in InferImplConv2D()
306 …Conv2DPadFunction(&output_hw, &pad_list, x_shape[h_axis], x_shape[w_axis], kernel_size, stride, di… in InferImplConv2D()
308 if (x_shape[h_axis] == Shape::SHP_ANY) { in InferImplConv2D()
314 …Conv2DPadFunction(&output_hw_min, &pad_list_min, x_min_shape[h_axis], x_min_shape[w_axis], kernel_… in InferImplConv2D()
316 …Conv2DPadFunction(&output_hw_max, &pad_list_max, x_max_shape[h_axis], x_max_shape[w_axis], kernel_… in InferImplConv2D()
|