Searched refs:output_hw_min (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore/core/ops/ |
| D | conv2d.cc | 210 std::vector<int64_t> output_hw_min; in Conv2dInferShape() local
216 …Conv2DPadFunction(&output_hw_min, &pad_list_min, x_min_shape[h_axis], x_min_shape[w_axis], kernel_… in Conv2dInferShape()
228 output_shape_min = {x_min_shape[n_axis], output_hw_min[0], output_hw_min[1], out_channel}; in Conv2dInferShape()
232 output_shape_min = {x_min_shape[n_axis], out_channel, output_hw_min[0], output_hw_min[1]}; in Conv2dInferShape()
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | prim_nn.cc | 302 std::vector<int64_t> output_hw_min; in InferImplConv2D() local
314 …Conv2DPadFunction(&output_hw_min, &pad_list_min, x_min_shape[h_axis], x_min_shape[w_axis], kernel_… in InferImplConv2D()
326 output_shape_min = {x_min_shape[n_axis], output_hw_min[0], output_hw_min[1], out_channel}; in InferImplConv2D()
330 output_shape_min = {x_min_shape[n_axis], out_channel, output_hw_min[0], output_hw_min[1]}; in InferImplConv2D()
|