| /third_party/mindspore/mindspore/ops/_op_impl/_custom_op/ |
| D | fused_abs_max1_impl.py | 35 def _update_tik(tik_instance, input_x_ub, broadcast_0_local_ub, block_index, res): argument
53 tik_instance.data_move(res[block_index, 0], broadcast_0_local_ub, 0, 1, 8, 0, 0)
83 with tik_instance.for_range(0, blocks, block_num=blocks) as block_index:
88 tik_instance.data_move(input_x_ub, input_x[each_block_element * block_index], 0, 1,
95 … tik_instance, res = _update_tik(tik_instance, input_x_ub, broadcast_0_local_ub, block_index, res)
106 with tik_instance.for_range(0, blocks, block_num=blocks) as block_index:
111 tik_instance.data_move(input_x_ub, input_x[512 * block_index], 0, 1, 512 // 8, 0, 0)
112 line_id = block_index % 19
120 … tik_instance, res = _update_tik(tik_instance, input_x_ub, broadcast_0_local_ub, block_index, res)
127 with tik_instance.for_range(0, blocks, block_num=blocks) as block_index:
[all …]
|
| D | img2col_impl.py | 42 with tik_instance.for_range(0, 32, block_num=32) as block_index:
47 … tik_instance.data_move(input_1_1_local_l1, input_x[block_index, 0, 0, 0, 0], 0, 1, 12544, 0, 0)
59 tik_instance.data_move(res[eeb1 * 9 + eeb0, i + 196 * block_index, 0, 0],
69 with tik_instance.for_range(0, 32, block_num=32) as block_index:
74 … tik_instance.data_move(input_1_1_local_l1, input_x[block_index, 0, 0, 0, 0], 0, 1, 25088, 0, 0)
86 tik_instance.data_move(res[eeb1 + eeb0 * 9, 49 * block_index + i, 0, 0],
97 with tik_instance.for_range(0, 32, block_num=32) as block_index:
102 … tik_instance.data_move(input_1_1_local_l1, input_x[block_index, 0, 0, 0, 0], 0, 1, 12544, 0, 0)
111 tik_instance.data_move(res[eeb, rep + block_index * 196, 0, 0],
121 with tik_instance.for_range(0, 32, block_num=32) as block_index:
[all …]
|
| D | matrix_combine_impl.py | 50 with tik_instance.for_range(0, blocks, block_num=blocks) as block_index:
53 tik_instance.data_move(input_x_ub, input_x[0, block_index * 2, 0], 0, 1, 16, 0, 0)
54 tik_instance.data_move(res[block_index * 2, 0], input_x_ub, 0, 1, 16, 0, 0)
61 with tik_instance.for_range(0, blocks, block_num=blocks) as block_index, \
80 … tik_instance.data_move(input_x_ub[j, 128 * i], input_x[i, block_index * tiling_dim + j, 0],
82 tik_instance.data_move(res[i * 128 + block_index * tiling_dim, 0], input_x_ub, 0, 1,
|
| D | cholesky_trsm_impl.py | 52 with tik_instance.for_range(0, blocks, block_num=blocks) as block_index:
58 …tik_instance.data_move(input_x_ub[i, 0], input_x[block_index * split_dim + i, block_index * split_…
108 …tik_instance.data_move(res[block_index, 0, 0], temp_ub, 0, 1, 8 * vector_repeat_times * split_dim,…
|
| D | matmul_cube_dense_right_impl.py | 67 with tik_instance.for_range(0, 32, block_num=32) as block_index:
68 core_m_idx = block_index // 16
69 core_n_idx = block_index % 16
|
| D | matmul_cube_dense_left_impl.py | 160 with tik_instance.for_range(0, 32, block_num=32) as block_index:
173 core_m_idx = block_index % 8
174 core_n_idx = block_index // 8
|
| /third_party/ffmpeg/libavcodec/ |
| D | vc1_block.c | 88 if (i > 3 ? v->mb_type[0][s->block_index[i] - s->block_wrap[i] - 1] : in vc1_put_blocks_clamped()
89 v->mb_type[0][s->block_index[i] - 2 * s->block_wrap[i] - 2]) { in vc1_put_blocks_clamped()
104 if (i > 3 ? v->mb_type[0][s->block_index[i] - s->block_wrap[i]] : in vc1_put_blocks_clamped()
105 v->mb_type[0][s->block_index[i] - 2 * s->block_wrap[i]]) { in vc1_put_blocks_clamped()
124 if (i > 3 ? v->mb_type[0][s->block_index[i] - 1] : in vc1_put_blocks_clamped()
125 v->mb_type[0][s->block_index[i] - 2]) { in vc1_put_blocks_clamped()
145 if (v->mb_type[0][s->block_index[i]]) { in vc1_put_blocks_clamped()
356 dc_val = s->dc_val[0] + s->block_index[n]; in vc1_i_pred_dc()
421 dc_val = s->dc_val[0] + s->block_index[n]; in ff_vc1_pred_dc()
480 xy = s->block_index[n]; in vc1_coded_block_pred()
[all …]
|
| D | vc1_loopfilter.c | 178 if (v->mb_type[0][s->block_index[i]] && v->mb_type[0][s->block_index[i] - 1]) in ff_vc1_p_overlap_filter()
191 if (s->mb_x && v->mb_type[0][s->block_index[i] - 2 + (i > 3)] && in ff_vc1_p_overlap_filter()
192 v->mb_type[0][s->block_index[i] - s->block_wrap[i] - 2 + (i > 3)]) in ff_vc1_p_overlap_filter()
195 if (v->mb_type[0][s->block_index[i]] && in ff_vc1_p_overlap_filter()
196 v->mb_type[0][s->block_index[i] - s->block_wrap[i]]) in ff_vc1_p_overlap_filter()
503 … &s->current_picture.motion_val[0][s->block_index[i] - 4 * s->b8_stride - 2 + v->blocks_off], in ff_vc1_p_loop_filter()
504 … i > 3 ? &v->mv_f[0][s->block_index[i] - 2 * s->mb_stride - 1 + v->mb_off] : in ff_vc1_p_loop_filter()
505 … &v->mv_f[0][s->block_index[i] - 4 * s->b8_stride - 2 + v->blocks_off], in ff_vc1_p_loop_filter()
523 … &s->current_picture.motion_val[0][s->block_index[i] - 4 * s->b8_stride + v->blocks_off], in ff_vc1_p_loop_filter()
524 … i > 3 ? &v->mv_f[0][s->block_index[i] - 2 * s->mb_stride + v->mb_off] : in ff_vc1_p_loop_filter()
[all …]
|
| D | vc1_pred.c | 241 xy = s->block_index[n]; in ff_vc1_pred_mv()
483 xy = s->block_index[n]; in ff_vc1_pred_mv_intfr()
536 pos_b = s->block_index[n_adj] - 2 * wrap; in ff_vc1_pred_mv_intfr()
540 B[0] = s->current_picture.motion_val[dir][s->block_index[n_adj] - 2 * wrap][0]; in ff_vc1_pred_mv_intfr()
541 B[1] = s->current_picture.motion_val[dir][s->block_index[n_adj] - 2 * wrap][1]; in ff_vc1_pred_mv_intfr()
543 …B[0] = (B[0] + s->current_picture.motion_val[dir][s->block_index[n_adj ^ 2] - 2 * wrap][0] + 1) >>… in ff_vc1_pred_mv_intfr()
544 …B[1] = (B[1] + s->current_picture.motion_val[dir][s->block_index[n_adj ^ 2] - 2 * wrap][1] + 1) >>… in ff_vc1_pred_mv_intfr()
551 pos_c = s->block_index[2] - 2 * wrap + 2; in ff_vc1_pred_mv_intfr()
555 … C[0] = s->current_picture.motion_val[dir][s->block_index[n_adj] - 2 * wrap + 2][0]; in ff_vc1_pred_mv_intfr()
556 … C[1] = s->current_picture.motion_val[dir][s->block_index[n_adj] - 2 * wrap + 2][1]; in ff_vc1_pred_mv_intfr()
[all …]
|
| D | vc1_mc.c | 111 int idx = v->mv_f[dir][s->block_index[0] + v->blocks_off] | in get_luma_mv()
112 (v->mv_f[dir][s->block_index[1] + v->blocks_off] << 1) | in get_luma_mv()
113 (v->mv_f[dir][s->block_index[2] + v->blocks_off] << 2) | in get_luma_mv()
114 (v->mv_f[dir][s->block_index[3] + v->blocks_off] << 3); in get_luma_mv()
143 int idx = !v->mb_type[0][s->block_index[0]] | in get_chroma_mv()
144 (!v->mb_type[0][s->block_index[1]] << 1) | in get_chroma_mv()
145 (!v->mb_type[0][s->block_index[2]] << 2) | in get_chroma_mv()
146 (!v->mb_type[0][s->block_index[3]] << 3); in get_chroma_mv()
199 s->current_picture.motion_val[1][s->block_index[i] + v->blocks_off][0] = mx; in ff_vc1_mc_1mv()
200 s->current_picture.motion_val[1][s->block_index[i] + v->blocks_off][1] = my; in ff_vc1_mc_1mv()
[all …]
|
| D | escape124.c | 144 unsigned block_index, depth; in decode_macroblock() local
157 block_index = get_bitsz(gb, depth); in decode_macroblock()
160 block_index += superblock_index << s->codebooks[1].depth; in decode_macroblock()
165 if (block_index >= s->codebooks[*codebook_index].size) in decode_macroblock()
168 return s->codebooks[*codebook_index].blocks[block_index]; in decode_macroblock()
|
| D | escape130.c | 204 block_index, block_x = 0; in escape130_decode_frame() local
235 for (block_index = 0; block_index < total_blocks; block_index++) { in escape130_decode_frame()
|
| D | mpegvideo.h | 293 int block_index[6]; ///< index to current MB in block based arrays with edges member
754 s->block_index[0]+=2; in ff_update_block_index()
755 s->block_index[1]+=2; in ff_update_block_index()
756 s->block_index[2]+=2; in ff_update_block_index()
757 s->block_index[3]+=2; in ff_update_block_index()
758 s->block_index[4]++; in ff_update_block_index()
759 s->block_index[5]++; in ff_update_block_index()
|
| D | mdec.c | 135 static const int block_index[6] = { 5, 4, 0, 1, 2, 3 }; in decode_mb() local
140 if ((ret = mdec_decode_block_intra(a, block[block_index[i]], in decode_mb()
141 block_index[i])) < 0) in decode_mb()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16_grad/ |
| D | pack_fp16_ext.c | 71 int real_cal_num, int block_index) { in RollingIm2ColPackUnitFp16() argument
96 int block_start = block_index + i; in RollingIm2ColPackUnitFp16()
116 int block_start = block_index + i; in RollingIm2ColPackUnitFp16()
138 int real_cal_num, int block_index) { in RollingCol2ImPackUnitFp16() argument
162 int output_col = (block_index + r) % output_w; in RollingCol2ImPackUnitFp16()
163 int output_row = (block_index + r) / output_w; in RollingCol2ImPackUnitFp16()
181 int output_col = (block_index + r) % output_w; in RollingCol2ImPackUnitFp16()
182 int output_row = (block_index + r) / output_w; in RollingCol2ImPackUnitFp16()
|
| D | pack_fp16_ext.h | 28 int real_cal_num, int block_index);
30 int real_cal_num, int block_index);
32 int real_cal_num, int block_index);
|
| /third_party/mesa3d/src/compiler/glsl/ |
| D | link_uniform_blocks.cpp | 224 unsigned *block_index,
240 unsigned *block_index, unsigned binding_offset, in process_block_array() argument
255 parcel, variables, b, block_index, in process_block_array()
259 parcel, variables, b, block_index, in process_block_array()
261 *block_index - first_index, ctx, prog); in process_block_array()
271 unsigned *block_index, unsigned binding_offset, in process_block_array_leaf() argument
275 unsigned i = *block_index; in process_block_array_leaf()
311 *block_index = *block_index + 1; in process_block_array_leaf()
|
| D | linker_util.cpp | 47 int block_index) in link_util_should_add_buffer_variable() argument
61 if (block_index != uniform->block_index || in link_util_should_add_buffer_variable()
|
| D | gl_nir_link_uniform_blocks.c | 560 unsigned block_index = 0; in link_linked_shader_uniform_blocks() local
575 fill_block(&blks[block_index], var, variables, &variable_index, in link_linked_shader_uniform_blocks()
577 block_index++; in link_linked_shader_uniform_blocks()
581 assert(block_index == *num_blocks); in link_linked_shader_uniform_blocks()
|
| D | gl_nir_linker.c | 506 int block_index = -1; in nir_build_program_resource_list() local
530 second_element_offset, block_index)) in nir_build_program_resource_list()
549 block_index = uniform->block_index; in nir_build_program_resource_list()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | layer_norm_grad_cpu_kernel.cc | 114 size_t block_index = c * thread_num2 + start; in LaunchKernel() local
118 for (size_t j = block_index * block_size_; j < (block_index + 1) * block_size_; ++j) { in LaunchKernel()
127 for (size_t j = block_index * block_size_; j < (block_index + 1) * block_size_; ++j) { in LaunchKernel()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32_grad/ |
| D | pack_ext.h | 28 int real_cal_num, int block_index);
30 int real_cal_num, int block_index);
|
| /third_party/mesa3d/src/compiler/spirv/ |
| D | vtn_variables.c | 321 nir_ssa_def *block_index = base->block_index; in vtn_pointer_dereference() local
348 if (!block_index || vtn_type_contains_block(b, type) || in vtn_pointer_dereference()
381 if (!block_index) { in vtn_pointer_dereference()
383 block_index = vtn_variable_resource_index(b, base->var, desc_arr_idx); in vtn_pointer_dereference()
385 block_index = vtn_resource_reindex(b, base->mode, in vtn_pointer_dereference()
386 block_index, desc_arr_idx); in vtn_pointer_dereference()
397 ptr->block_index = block_index; in vtn_pointer_dereference()
406 nir_ssa_def *desc = vtn_descriptor_load(b, base->mode, block_index); in vtn_pointer_dereference()
583 if (!ptr->block_index) { in vtn_pointer_to_descriptor()
590 vtn_assert(ptr->deref == NULL && ptr->block_index != NULL); in vtn_pointer_to_descriptor()
[all …]
|
| /third_party/gstreamer/gstplugins_good/gst/audioparsers/ |
| D | gstwavpackparse.c | 416 hdl &= gst_byte_reader_get_uint32_le (&br, &wph.block_index); in gst_wavpack_parse_frame_header()
428 GST_LOG_OBJECT (parse, "block index %d", wph.block_index); in gst_wavpack_parse_frame_header()
433 if (!parse->total_samples && wph.block_index == 0 && wph.total_samples != -1) { in gst_wavpack_parse_frame_header()
551 gst_util_uint64_scale_int (wph.block_index, GST_SECOND, rate); in gst_wavpack_parse_handle_frame()
554 gst_util_uint64_scale_int (wph.block_index + wph.block_samples, in gst_wavpack_parse_handle_frame()
|
| /third_party/gstreamer/gstplugins_good/ext/wavpack/ |
| D | gstwavpackenc.c | 589 enc->pending_offset = wph.block_index; in gst_wavpack_enc_push_block()
590 } else if (enc->pending_offset == wph.block_index) { in gst_wavpack_enc_push_block()
596 enc->pending_offset = wph.block_index; in gst_wavpack_enc_push_block()
615 if (wph.block_index == 0) { in gst_wavpack_enc_push_block()
634 GST_BUFFER_OFFSET (buffer) = wph.block_index; in gst_wavpack_enc_push_block()
635 GST_BUFFER_OFFSET_END (buffer) = wph.block_index + wph.block_samples; in gst_wavpack_enc_push_block()
|