Home
last modified time | relevance | path

Searched refs:bs_idx (Results 1 – 8 of 8) sorted by relevance

/third_party/mindspore/mindspore/ops/_op_impl/_custom_op/
Ddsd_impl.py77 bs_idx = channel_idx % batch_size
99 tik_inst.data_move(v_global_l1[0, 0, 0, 0], v_gm[bs_idx * seq_len // 16 + global_idx,
114 w2_gm[bs_idx, head_idx, w_idx, cpt_idx *
137 v_gm[bs_idx * seq_len//16 + w_idx * 4, head_idx *
145 … tik_inst.data_move(w1_l1[brick_i, 0, 0, 0], w1_gm[bs_idx, head_idx, w_idx, 0, brick_i, 0, 0], 0,
155 …tik_inst.data_move(output_gm[bs_idx, head_idx, 0, w_idx*(block_size//16), 0, 0], output_ub[0, 0, 0…
Ddsd_back_impl.py126 bs_idx = channel_idx % batch_size
137 … tik_inst.data_move(d_a_l1[0, brick_i, 0, 0], d_a_gm[bs_idx, head_idx, brick_i, 0, 0, 0], 0,
167 … tik_inst.data_move(w_local_l1[0, 0, 0, 0], w1_gm[bs_idx, head_idx, w_idx, 0, 0, 0, 0], 0,
197 … tik_inst.data_move(w_global_l1[0, 0, 0, 0], w2_gm[bs_idx, head_idx, w_idx_1, w_idx, 0, 0, 0], 0,
229 tik_inst.data_move(d_v_gm[bs_idx*seq_len//16+w_idx * (block_size // 16) + h_idx,
261 v_gm[bs_idx*seq_len//16+w_idx *
280 … tik_inst.data_move(d_w1_gm[bs_idx, head_idx, w_idx, 0, 0, 0, 0], d_w_local_ub[0, 0, 0, 0], 0,
291 v_gm[bs_idx*seq_len//16 + (
349 … tik_inst.data_move(d_w2_gm[bs_idx, head_idx, w_idx, 0, 0, 0, 0], d_w_global_ub[0, 0, 0, 0], 0,
358 … tik_inst.data_move(d_w2_gm[bs_idx, head_idx, w_idx_i, 0, h_idx + m_idx, 0, 0],
/third_party/mesa3d/src/gallium/drivers/radeon/
Dradeon_vce_50.c73 unsigned dep, bs_idx = enc->bs_idx++; in encode() local
77 if (bs_idx == 0) in encode()
86 enc->task_info(enc, 0x00000003, dep, 0, bs_idx); in encode()
92 bs_offset = -(signed)(bs_idx * enc->bs_size); in encode()
Dradeon_vce_52.c216 unsigned dep, bs_idx = enc->bs_idx++; in encode() local
220 if (bs_idx == 0) in encode()
229 enc->task_info(enc, 0x00000003, dep, 0, bs_idx); in encode()
235 bs_offset = -(signed)(bs_idx * enc->bs_size); in encode()
Dradeon_vce.c56 enc->bs_idx = 0; in flush()
331 if (!enc->dual_inst || enc->bs_idx > 1) in rvce_end_frame()
Dradeon_vce.h413 unsigned bs_idx; member
/third_party/mesa3d/src/gallium/drivers/r600/
Dradeon_vce.c68 enc->bs_idx = 0; in flush()
340 if (!enc->dual_inst || enc->bs_idx > 1) in rvce_end_frame()
Dradeon_vce.h411 unsigned bs_idx; member