Searched refs:w_axis (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore/core/ops/ |
| D | conv2d.cc | 174 uint64_t w_axis = 3; in Conv2dInferShape() local
179 w_axis = 2; in Conv2dInferShape()
200 if ((w_shape[w_axis] != Shape::SHP_ANY) && (w_shape[w_axis] != kernel_size[1])) { in Conv2dInferShape()
201 …MS_LOG(EXCEPTION) << "weight width = " << w_shape[w_axis] << ", must equal to = " << kernel_size[1… 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 | 266 uint64_t w_axis = 3; in InferImplConv2D() local
271 w_axis = 2; in InferImplConv2D()
289 if ((w_shape[w_axis] != Shape::SHP_ANY) && (w_shape[w_axis] != kernel_size[1])) { in InferImplConv2D()
290 …MS_LOG(EXCEPTION) << "weight width = " << w_shape[w_axis] << ", must equal to = " << kernel_size[1… in InferImplConv2D()
306 …Conv2DPadFunction(&output_hw, &pad_list, x_shape[h_axis], x_shape[w_axis], kernel_size, stride, di… in InferImplConv2D()
311 if (x_shape[w_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()
|