Home
last modified time | relevance | path

Searched refs:block_w (Results 1 – 24 of 24) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/base/
Dbatch_to_space_base.c22 int block_w = block[1]; in BatchToSpaceNoCropForNHWC() local
26 int stride_h = block_w * out_n; in BatchToSpaceNoCropForNHWC()
37 for (int bw = 0; bw < block_w; ++bw) { in BatchToSpaceNoCropForNHWC()
51 int block_w = block[1]; in BatchToSpaceForNHWC() local
52 if (block_h == 0 || block_w == 0) { in BatchToSpaceForNHWC()
62 int w_start = crops[2] / block_w; in BatchToSpaceForNHWC()
64 int w_end = MSMIN((in_w * block_w - crops[3]) / block_w + 1, in_w); in BatchToSpaceForNHWC()
65 int w_valid_end = in_w * block_w - crops[3] - 1; in BatchToSpaceForNHWC()
67 int stride_h = block_w * out_n; in BatchToSpaceForNHWC()
82 for (int bw = 0; bw < block_w; ++bw) { in BatchToSpaceForNHWC()
[all …]
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/
Dspace_to_batch_nd_infer.c36 int block_w = 1; in SpaceSetOutputShapeFromParam() local
40 block_w = block_shape[1]; in SpaceSetOutputShapeFromParam()
42 if (input->shape_[kNHWC_N] == 0 || block_shape[0] * block_w > INT_MAX / input->shape_[kNHWC_N]) { in SpaceSetOutputShapeFromParam()
45 outputs[0]->shape_[kNHWC_N] = input->shape_[kNHWC_N] * block_shape[0] * block_w; in SpaceSetOutputShapeFromParam()
49 if (block_shape[0] == 0 || block_w == 0) { in SpaceSetOutputShapeFromParam()
56 outputs[0]->shape_[kNHWC_W] = (input->shape_[kNHWC_W] + padding_left + padding_right) / block_w; in SpaceSetOutputShapeFromParam()
77 int block_w = 1; in SpaceSetOutputShapeFromInput() local
81 block_w = block_shape[1]; in SpaceSetOutputShapeFromInput()
85 if (input->shape_[kNHWC_N] == 0 || block_shape[0] * block_w > INT_MAX / input->shape_[kNHWC_N]) { in SpaceSetOutputShapeFromInput()
88 output_shape[kNHWC_N] = input->shape_[kNHWC_N] * block_shape[0] * block_w; in SpaceSetOutputShapeFromInput()
[all …]
Dspace_to_batch_infer.c45 int block_w = 1; in SpaceToBatchInferShape() local
49 block_w = block_shape[1]; in SpaceToBatchInferShape()
53 NNACL_CHECK_ZERO_RETURN_ERR(block_w); in SpaceToBatchInferShape()
54 MS_CHECK_INT_MUL_NOT_OVERFLOW(block_shape[0], block_w, NNACL_ERR); in SpaceToBatchInferShape()
55 MS_CHECK_INT_MUL_NOT_OVERFLOW(input->shape_[kNHWC_N], block_shape[0] * block_w, NNACL_ERR); in SpaceToBatchInferShape()
56 outputs[0]->shape_[kNHWC_N] = input->shape_[kNHWC_N] * (block_shape[0] * block_w); in SpaceToBatchInferShape()
58 outputs[0]->shape_[kNHWC_W] = (input->shape_[kNHWC_W] + padding_left + padding_right) / block_w; in SpaceToBatchInferShape()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/
Dbatch_to_space_int8.c22 int block_w = block[1]; in BatchToSpaceNoCropForNHWCInt8() local
26 size_t stride_h = block_w * out_n; in BatchToSpaceNoCropForNHWCInt8()
41 for (int bw = 0; bw < block_w; ++bw) { in BatchToSpaceNoCropForNHWCInt8()
59 int block_w = block[1]; in BatchToSpaceForNHWCInt8() local
67 int w_start = crops[2] / block_w; in BatchToSpaceForNHWCInt8()
69 int w_end = MSMIN((in_w * block_w - crops[3]) / block_w + 1, in_w); in BatchToSpaceForNHWCInt8()
70 int w_valid_end = in_w * block_w - crops[3] - 1; in BatchToSpaceForNHWCInt8()
72 size_t stride_h = block_w * out_n; in BatchToSpaceForNHWCInt8()
92 for (int bw = 0; bw < block_w; ++bw) { in BatchToSpaceForNHWCInt8()
93 size_t w_index = w * block_w + bw; in BatchToSpaceForNHWCInt8()
Dspace_to_batch_int8.c25 int block_w = block_sizes[1]; in DoSpaceToBatchNHWCInt8() local
35 NNACL_CHECK_ZERO_RETURN(block_w); in DoSpaceToBatchNHWCInt8()
38 int32_t stride_w = (n / in_shape[0]) % block_w; in DoSpaceToBatchNHWCInt8()
39 int32_t stride_h = (n / in_shape[0]) / block_w; in DoSpaceToBatchNHWCInt8()
44 size_t in_offset2 = in_offset1 + (w * block_w + stride_w) * in_strides[2]; in DoSpaceToBatchNHWCInt8()
/third_party/ffmpeg/libavcodec/
Dvideodsp_template.c26 int block_w, int block_h, in FUNC()
35 av_assert2(block_w * sizeof(pixel) <= FFABS(buf_linesize)); in FUNC()
50 } else if (src_x <= -block_w) { in FUNC()
51 src += (1 - block_w - src_x) * sizeof(pixel); in FUNC()
52 src_x = 1 - block_w; in FUNC()
58 end_x = FFMIN(block_w, w-src_x); in FUNC()
60 av_assert2(start_x < end_x && block_w); in FUNC()
96 for (x = end_x; x < block_w; x++) { in FUNC()
Dsnowenc.c242 const int block_w= 1<<(LOG2_MB_SIZE - level); in encode_q_branch() local
259 uint8_t *current_data[3]= { s->input_picture->data[0] + (x + y* stride)*block_w, in encode_q_branch()
260 …s->input_picture->data[1] + ((x*block_w)>>s->chroma_h_shift) + ((y*uvstride*block_w)>>s->chroma_v_… in encode_q_branch()
261 …s->input_picture->data[2] + ((x*block_w)>>s->chroma_h_shift) + ((y*uvstride*block_w)>>s->chroma_v_… in encode_q_branch()
308 c->xmin = - x*block_w - 16+3; in encode_q_branch()
309 c->ymin = - y*block_w - 16+3; in encode_q_branch()
310 c->xmax = - (x+1)*block_w + (w<<(LOG2_MB_SIZE - s->block_max_depth)) + 16-3; in encode_q_branch()
311 c->ymax = - (y+1)*block_w + (h<<(LOG2_MB_SIZE - s->block_max_depth)) + 16-3; in encode_q_branch()
335 init_ref(c, current_data, s->last_picture[ref]->data, NULL, block_w*x, block_w*y, 0); in encode_q_branch()
338 (1<<16)>>shift, level-LOG2_MB_SIZE+4, block_w); in encode_q_branch()
[all …]
Dvideodsp.h35 int block_w, int block_h,\
66 int block_w, int block_h,
Dsnow.h411 int block_w = plane_index ? block_size>>s->chroma_h_shift : block_size; in predict_slice() local
446 block_w*mb_x - block_w/2, in predict_slice()
448 block_w, block_h, in predict_slice()
467 const int block_w= 1<<rem_depth; in set_blocks() local
482 for(i=0; i<block_w; i++){ in set_blocks()
Dsnowdec.c41 int block_w = plane_index ? block_size>>s->chroma_h_shift : block_size; in predict_slice_buffered() local
82 block_w*mb_x - block_w/2, in predict_slice_buffered()
84 block_w, block_h, in predict_slice_buffered()
103 avmv->w = block_w; in predict_slice_buffered()
105 avmv->dst_x = block_w*mb_x - block_w/2; in predict_slice_buffered()
Dhevcdec.c1482 int block_w, int block_h, int luma_weight, int luma_offset) in luma_mc_uni() argument
1493 int idx = ff_hevc_pel_weight[block_w]; in luma_mc_uni()
1500 x_off >= pic_width - block_w - QPEL_EXTRA_AFTER || in luma_mc_uni()
1508 block_w + QPEL_EXTRA, in luma_mc_uni()
1518 block_h, mx, my, block_w); in luma_mc_uni()
1522 luma_weight, luma_offset, mx, my, block_w); in luma_mc_uni()
1543 … int block_w, int block_h, AVFrame *ref1, const Mv *mv1, struct MvField *current_mv) in luma_mc_bi() argument
1560 int idx = ff_hevc_pel_weight[block_w]; in luma_mc_bi()
1566 x_off0 >= pic_width - block_w - QPEL_EXTRA_AFTER || in luma_mc_bi()
1574 block_w + QPEL_EXTRA, in luma_mc_bi()
[all …]
Dfmvc.c499 int block_w = s->blocks[block].w; in decode_frame() local
506 for (l = 0; l < block_w; l++) in decode_frame()
511 dst = &row[block_w]; in decode_frame()
Dvp8.c1811 int x_off, int y_off, int block_w, int block_h, in vp8_mc_luma() argument
1829 if (x_off < mx_idx || x_off >= width - block_w - subpel_idx[2][mx] || in vp8_mc_luma()
1834 block_w + subpel_idx[1][mx], in vp8_mc_luma()
1869 int x_off, int y_off, int block_w, int block_h, in vp8_mc_chroma() argument
1886 if (x_off < mx_idx || x_off >= width - block_w - subpel_idx[2][mx] || in vp8_mc_chroma()
1891 block_w + subpel_idx[1][mx], in vp8_mc_chroma()
1900 block_w + subpel_idx[1][mx], in vp8_mc_chroma()
1919 int bx_off, int by_off, int block_w, int block_h, in vp8_mc_part() argument
1927 block_w, block_h, width, height, s->linesize, in vp8_mc_part()
1928 s->put_pixels_tab[block_w == 8]); in vp8_mc_part()
[all …]
/third_party/ffmpeg/libavcodec/x86/
Dvideodsp_init.c152 x86_reg block_w, x86_reg block_h, in emulated_edge_mc() argument
165 av_assert2(block_w <= FFABS(dst_stride)); in emulated_edge_mc()
179 } else if (src_x <= -block_w) { in emulated_edge_mc()
180 src += 1 - block_w - src_x; in emulated_edge_mc()
181 src_x = 1 - block_w; in emulated_edge_mc()
187 end_x = FFMIN(block_w, w-src_x); in emulated_edge_mc()
188 av_assert2(start_x < end_x && block_w > 0); in emulated_edge_mc()
213 p = block_w - end_x; in emulated_edge_mc()
229 int block_w, int block_h, in emulated_edge_mc_mmx() argument
232 emulated_edge_mc(buf, src, buf_stride, src_stride, block_w, block_h, in emulated_edge_mc_mmx()
[all …]
/third_party/ffmpeg/libavfilter/
Dvf_shufflepixels.c38 int block_w, block_h; member
92 width = FFMIN(s->block_w, s->planewidth[0] - x); in make_horizontal_map()
93 map[rand * s->block_w] = x; in make_horizontal_map()
95 width = FFMIN(s->block_w, s->planewidth[0] - rand * s->block_w); in make_horizontal_map()
96 map[x] = rand * s->block_w; in make_horizontal_map()
102 map[rand * s->block_w + i] = map[rand * s->block_w] + i; in make_horizontal_map()
157 int nb_blocks_w = s->planewidth[0] / s->block_w; in make_block_map()
168 int in = yin * s->block_h * s->planewidth[0] + xin * s->block_w; in make_block_map()
171 int out = yout * s->block_h * s->planewidth[0] + xout * s->block_w; in make_block_map()
182 for (int x = 0; x < s->block_w; x++) { in make_block_map()
[all …]
/third_party/mesa3d/src/mesa/main/
Dtexcompress_astc.cpp545 Decoder(int block_w, int block_h, int block_d, bool srgb, bool output_unorm8) in Decoder() argument
546 : block_w(block_w), block_h(block_h), block_d(block_d), srgb(srgb), in Decoder()
551 int block_w, block_h, block_d; member in Decoder
640 void compute_infill_weights(int block_w, int block_h, int block_d);
656 for (int i = 0; i < block_w * block_h * block_d; ++i) { in decode()
1263 void Block::compute_infill_weights(int block_w, int block_h, int block_d) in compute_infill_weights() argument
1265 int Ds = block_w <= 1 ? 0 : (1024 + block_w / 2) / (block_w - 1); in compute_infill_weights()
1270 for (int s = 0; s < block_w; ++s) { in compute_infill_weights()
1311 infill_weights[0][s + t*block_w + r*block_w*block_h] = i0; in compute_infill_weights()
1312 infill_weights[1][s + t*block_w + r*block_w*block_h] = i1; in compute_infill_weights()
[all …]
/third_party/mesa3d/src/gallium/drivers/lima/
Dlima_context.h38 int block_w, block_h; member
165 uint16_t block_w, block_h; member
Dlima_gpu.h127 #define PLBU_CMD_BLOCK_STRIDE(block_w) PLBU_CMD((block_w) & 0xff, 0x30000000) argument
Dlima_job.c82 fb->block_w = width; in lima_get_fb_info()
459 PLBU_CMD_BLOCK_STRIDE(fb->block_w); in lima_pack_head_plbu_cmd()
463 fb->block_w * fb->block_h); in lima_pack_head_plbu_cmd()
579 int offset = ((y >> fb->shift_h) * fb->block_w + in lima_generate_pp_stream()
662 .block_w = fb->block_w, in lima_update_damage_pp_stream()
697 s->key.block_w = fb->block_w; in lima_update_damage_pp_stream()
Dlima_job.h58 int block_w, block_h; member
/third_party/mindspore/mindspore/_extends/graph_kernel/model/
Dgraph_parallel.py101 block_w = (weight + block_x - 1) // block_x
105 self.block_weight = (block_w + all_reduce *
/third_party/ffmpeg/tests/checkasm/
Dvideodsp.c73 int block_w, int block_h, \
/third_party/mesa3d/src/broadcom/vulkan/
Dv3dv_meta_copy.c406 const uint32_t block_w = vk_format_get_blockwidth(image->vk.format); in copy_image_to_buffer_tlb() local
408 const uint32_t width = DIV_ROUND_UP(region->imageExtent.width, block_w); in copy_image_to_buffer_tlb()
812 const uint32_t block_w = vk_format_get_blockwidth(src->vk.format); in copy_image_tfu() local
814 uint32_t width = DIV_ROUND_UP(region->extent.width, block_w); in copy_image_tfu()
914 const uint32_t block_w = vk_format_get_blockwidth(dst->vk.format); in copy_image_tlb() local
916 const uint32_t width = DIV_ROUND_UP(region->extent.width, block_w); in copy_image_tlb()
1302 const uint32_t block_w = vk_format_get_blockwidth(image->vk.format); in copy_buffer_to_image_tfu() local
1304 width = DIV_ROUND_UP(width, block_w); in copy_buffer_to_image_tfu()
1418 const uint32_t block_w = vk_format_get_blockwidth(image->vk.format); in copy_buffer_to_image_tlb() local
1420 const uint32_t width = DIV_ROUND_UP(region->imageExtent.width, block_w); in copy_buffer_to_image_tlb()
[all …]
/third_party/mesa3d/src/panfrost/lib/
Dpan_texture.c677 unsigned block_w = util_format_get_blockwidth(iview->format); in panfrost_needs_explicit_stride() local
682 DIV_ROUND_UP(u_minify(iview->image->layout.width, l), block_w) * in panfrost_needs_explicit_stride()