Home
last modified time | relevance | path

Searched refs:out_h (Results 1 – 25 of 47) sorted by relevance

12

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Dspace_to_batch_fp32.c45 for (int out_h = 0; out_h < output_height; out_h++) { in DoSpaceToBatch() local
47 float *out = output + out_b * out_stride[0] + out_h * out_stride[1] + out_w * out_stride[2]; in DoSpaceToBatch()
48 if (out_h * block_shape_height + shift_h < padding_top || in DoSpaceToBatch()
49 out_h * block_shape_height + shift_h >= padding_top + input_height || in DoSpaceToBatch()
54 int in_h = (out_h * block_shape_height + shift_h) - padding_top; in DoSpaceToBatch()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16_grad/
Dconvolution_grad_filter.c35 int out_h = conv_param->output_h_; in FilterGrad32Arm() local
38 int m = out_h * out_w; in FilterGrad32Arm()
40 int y_size = out_ch * out_h * out_w; in FilterGrad32Arm()
113 int out_h = conv_param->output_h_; in FilterGrad16Arm() local
116 int m = out_h * out_w; in FilterGrad16Arm()
118 int y_size = out_ch * out_h * out_w; in FilterGrad16Arm()
171 int out_h = conv_param->output_h_; in FilterGrad8Arm() local
174 int m = out_h * out_w; in FilterGrad8Arm()
176 int y_size = out_ch * out_h * out_w; in FilterGrad8Arm()
220 int out_h = conv_param->output_h_; in FilterGrad4Arm() local
[all …]
Dconvolution_grad_input.c28 int out_h = conv_param->output_h_; in ConvDwInputGrad16() local
37 int out_size = out_h * out_w * out_ch; in ConvDwInputGrad16()
133 int out_h = conv_param->output_h_; in ConvDwInputGrad8() local
142 int out_size = out_h * out_w * out_ch; in ConvDwInputGrad8()
216 int out_h = conv_param->output_h_; in ConvDwInputGrad4() local
225 int out_size = out_h * out_w * out_ch; in ConvDwInputGrad4()
292 int out_h = conv_param->output_h_; in ConvDwInputGradFp16() local
302 int out_size = out_h * out_w * out_ch; in ConvDwInputGradFp16()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32_grad/
Dconvolution_grad_filter.c33 int out_h = conv_param->output_h_; in FilterGrad16Arm() local
36 int m = out_h * out_w; in FilterGrad16Arm()
38 int y_size = out_ch * out_h * out_w; in FilterGrad16Arm()
110 int out_h = conv_param->output_h_; in FilterGrad12Arm() local
113 int m = out_h * out_w; in FilterGrad12Arm()
115 int y_size = out_ch * out_h * out_w; in FilterGrad12Arm()
180 int out_h = conv_param->output_h_; in FilterGrad8Arm() local
183 int m = out_h * out_w; in FilterGrad8Arm()
185 int y_size = out_ch * out_h * out_w; in FilterGrad8Arm()
239 int out_h = conv_param->output_h_; in FilterGrad4Arm() local
[all …]
/third_party/mindspore/tests/ut/cpp/python_input/gtest_input/pynative/
Dops_test.py37 …out_h = (height + pad_top + pad_bottom - filter_h - (filter_h - 1) * (dilation[2] - 1)) // stride[…
41 col = np.zeros((batch_num, channel, filter_h, filter_w, out_h, out_w)).astype(img.dtype)
44 y_max = y + stride[2] * out_h
49 col = col.transpose(0, 4, 5, 1, 2, 3).reshape(batch_num * out_h * out_w, -1)
70 …out_h = 1 + int((x_h + pad_top + pad_bottom - filter_h - (filter_h - 1) * (dilation[2] - 1)) / str…
75 out = out.reshape(batch_num, out_h, out_w, -1).transpose(0, 3, 1, 2)
/third_party/mesa3d/src/vulkan/util/
Dvk_extensions_gen.py211 extensions, out_c, out_h, includes = []): argument
229 if out_h:
230 with open(out_h, 'w') as f:
254 extensions, args.out_c, args.out_h, [])
Dvk_entrypoints_gen.py222 assert os.path.dirname(args.out_c) == os.path.dirname(args.out_h)
227 'header': os.path.basename(args.out_h),
240 with open(args.out_h, 'w') as f:
241 guard = os.path.basename(args.out_h).replace('.', '_').upper()
Dvk_cmd_queue_gen.py406 assert os.path.dirname(args.out_c) == os.path.dirname(args.out_h)
409 'header': os.path.basename(args.out_h),
428 with open(args.out_h, 'wb') as f:
429 guard = os.path.basename(args.out_h).replace('.', '_').upper()
/third_party/mindspore/tests/vm_impl/
Dvm_me.py36 out_h = (height - pool_h) // stride + 1
43 out = out.reshape((num, out_h, out_w, channel)).transpose(0, 3, 1, 2)
187 out_h = (height + pad_top + pad_bottom - filter_h) // stride_h + 1
189 col = col.reshape(batch_num, out_h, out_w, channel, filter_h, filter_w) \
198 y_max = y + stride_h * out_h
273 …out_h = 1 + int((x_h + pad_top + pad_bottom - filter_h - (filter_h - 1) * (dilation_h - 1)) / stri…
278 out = out.reshape((batch_num, out_h, out_w, -1)).transpose(0, 3, 1, 2)
388 …out_h = (height + pad_top + pad_bottom - filter_h - (filter_h - 1) * (dilation_h - 1)) // stride_h…
392 col = np.zeros((batch_num, channel, filter_h, filter_w, out_h, out_w)).astype(img.dtype)
395 y_max = y + stride_h * out_h
[all …]
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/base/
Dconv_common_base.c50 int out_h = conv_param->output_h_; in SelectOutputUnit() local
55 int unit2 = UP_DIV(out_w * out_h, C12NUM * conv_param->op_parameter_.thread_num_); in SelectOutputUnit()
62 float common_cost = (float)out_h * out_w * in_c * out_c * kernel_h * kernel_w; in SelectOutputUnit()
71 UP_DIV(out_w, i) * UP_DIV(out_h, i); in SelectOutputUnit()
/third_party/toybox/kconfig/
Dconfdata.c693 FILE *out, *out_h; in conf_write_autoconf() local
708 out_h = fopen(".tmpconfig.h", "w"); in conf_write_autoconf()
709 if (!out_h) { in conf_write_autoconf()
723 fprintf(out_h, "/*\n" in conf_write_autoconf()
743 fprintf(out_h, "#define CONFIG_%s_MODULE 1\n", sym->name); in conf_write_autoconf()
747 fprintf(out_h, "#define "CONFIG_PREFIX"%s 1\n", sym->name); in conf_write_autoconf()
754 fprintf(out_h, "#define "CONFIG_PREFIX"%s \"", sym->name); in conf_write_autoconf()
759 fwrite(str, l, 1, out_h); in conf_write_autoconf()
765 fprintf(out_h, "\\%c", *str); in conf_write_autoconf()
769 fputs("\"\n", out_h); in conf_write_autoconf()
[all …]
/third_party/mindspore/mindspore/core/ops/fusion/
Davg_pool_fusion.cc80 int64_t out_h = abstract::Shape::SHP_ANY; in InferShape() local
83 out_h = static_cast<int64_t>(ceil((in_h - (kernel_h - 1)) / static_cast<float>(stride_h))); in InferShape()
86 out_h = static_cast<int64_t>(ceil(in_h / static_cast<float>(stride_h))); in InferShape()
89 std::vector<int64_t> out_shape = {batch, channel, out_h, out_w}; in InferShape()
91 out_shape = {batch, out_h, out_w, channel}; in InferShape()
Dmax_pool_fusion.cc77 int64_t out_h = abstract::Shape::SHP_ANY; in InferShape() local
80 out_h = static_cast<int64_t>(ceil((in_h - (kernel_h - 1)) / static_cast<float>(stride_h))); in InferShape()
83 out_h = static_cast<int64_t>(ceil(in_h / static_cast<float>(stride_h))); in InferShape()
86 std::vector<int64_t> out_shape = {batch, channel, out_h, out_w}; in InferShape()
88 out_shape = {batch, out_h, out_w, channel}; in InferShape()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/
Dresize_nearest_neighbor_impl.cu37 int out_h; in ResizeNearestNeighbor() local
44 out_h = pos_array[2]; in ResizeNearestNeighbor()
47 …min((align_corners) ? static_cast<int>(roundf(out_h * h_scale)) : static_cast<int>(floorf(out_h * … in ResizeNearestNeighbor()
/third_party/mindspore/mindspore/core/ops/
Dmax_pool.cc106 int64_t out_h = abstract::Shape::SHP_ANY; in InferShape() local
109 out_h = static_cast<int64_t>(ceil((in_h - (kernel_h - 1)) + static_cast<float>(stride_h) - 1) / in InferShape()
114 out_h = static_cast<int64_t>(ceil(in_h / static_cast<float>(stride_h))); in InferShape()
117 std::vector<int64_t> out_shape = {batch, channel, out_h, out_w}; in InferShape()
119 out_shape = {batch, out_h, out_w, channel}; in InferShape()
Davg_pool.cc111 int64_t out_h = abstract::Shape::SHP_ANY; in InferShape() local
114 out_h = static_cast<int64_t>(std::ceil((in_h - (kernel_h - 1)) / static_cast<float>(stride_h))); in InferShape()
117 out_h = static_cast<int64_t>(std::ceil(in_h / static_cast<float>(stride_h))); in InferShape()
120 std::vector<int64_t> out_shape = {batch, channel, out_h, out_w}; in InferShape()
122 out_shape = {batch, out_h, out_w, channel}; in InferShape()
Dconv2d.cc77 int64_t out_h = -1; in Conv2DPadFunction() local
80 out_h = in Conv2DPadFunction()
82 if (is_min_shape && out_h < 1) { in Conv2DPadFunction()
83 out_h = 1L; in Conv2DPadFunction()
93 output_hw->push_back(out_h); in Conv2DPadFunction()
123 int64_t out_h = -1; in Conv2DPadFunction() local
126 out_h = static_cast<int64_t>(std::floor( in Conv2DPadFunction()
129 if (is_min_shape && out_h < 1) { in Conv2DPadFunction()
130 out_h = 1L; in Conv2DPadFunction()
141 output_hw->push_back(out_h); in Conv2DPadFunction()
Davg_pool_3d.cc65 int64_t out_h = 0; in GetOutputShape() local
70 out_h = in GetOutputShape()
77 if ((out_h - 1) * stride_h >= in_h + pad_list[2]) { in GetOutputShape()
78 out_h--; in GetOutputShape()
85 …out_h = static_cast<int64_t>(std::floor((in_h + pad_list[2] + pad_list[3] - kernel_h) / stride_h +… in GetOutputShape()
88 std::vector<int64_t> output_shape = {in_shape[0], in_shape[1], out_d, out_h, out_w}; in GetOutputShape()
/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/common/
Dpack_tests.cc69 int out_h = conv_param->output_h_; in TEST_F() local
74 int output_count = out_h * out_w; in TEST_F()
132 int out_h = conv_param->output_h_; in TEST_F() local
137 int output_count = out_h * out_w; in TEST_F()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/
Dspace_to_batch_int8.c59 int out_h = param->output_shape_[1]; in DoSpaceToBatchPaddingNHWCInt8() local
71 int out_b_offset = i * out_h * out_w * channel; in DoSpaceToBatchPaddingNHWCInt8()
72 for (int j = 0; j < out_h; ++j) { in DoSpaceToBatchPaddingNHWCInt8()
Dresize_int8.h30 …8(const int8_t *input_ptr, int8_t *output_ptr, int batch, int in_h, int in_w, int out_h, int out_w,
34 int out_h, int out_w, int channel, int index, int count,
Dresize_int8.c22 …8(const int8_t *input_ptr, int8_t *output_ptr, int batch, int in_h, int in_w, int out_h, int out_w, in ResizeBilinearInt8() argument
28 int out_plane = out_h * out_w; in ResizeBilinearInt8()
71 int out_h, int out_w, int channel, int index, int count, in ResizeBilinearWithFloatScaleInt8() argument
77 int out_plane = out_h * out_w; in ResizeBilinearWithFloatScaleInt8()
/third_party/mindspore/mindspore/_extends/graph_kernel/expanders/
Dconv2d.py139 out_h, out_w = (h0 - h1) // stride[-2] + 1, (w0 - w1) // stride[-1] + 1
140 if ((n0 * out_h * out_w) % OUT_NHW_ALIGN) != 0:
142 n0, out_h, out_w, OUT_NHW_ALIGN))
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32_grad/
Ddeconvolution_grad_filter.cc95 int out_h = conv_param->output_h_; in Execute() local
114 float *im = dy_addr + (i * (out_h * out_w * out_ch) + j * (out_ch / groups)); in Execute()
Dconvolution_grad_input.cc99 int out_h = conv_param->output_h_; in Execute() local
102 int m = out_h * out_w; in Execute()

12