Home
last modified time | relevance | path

Searched refs:head_idx (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore/ops/_op_impl/_custom_op/
Ddsd_impl.py76 head_idx = channel_idx // batch_size
98 global_idx = 3 - head_idx % 4
100 … head_idx * v_embedding // 16, 0, 0], 0, seq_len // (4 * 16),
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.py125 head_idx = channel_idx // batch_size
127 global_idx = 3 - head_idx % 4
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,
230 head_idx*v_embedding//16, 0, 0],
262 … (head_size//16)+brick_i, head_idx*v_embedding//16, 0, 0],
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,
292 … w_idx * (block_size//16) + global_idx), head_idx * v_embedding//16, 0, 0],
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,
[all …]
/third_party/mesa3d/src/util/
Dsparse_array.c270 uint32_t head_idx = current_head; /* Index is the bottom 32 bits */ in util_sparse_array_free_list_pop_idx() local
271 void *head_elem = util_sparse_array_get(fl->arr, head_idx); in util_sparse_array_free_list_pop_idx()
276 return head_idx; in util_sparse_array_free_list_pop_idx()
291 uint32_t head_idx = current_head; /* Index is the bottom 32 bits */ in util_sparse_array_free_list_pop_elem() local
292 void *head_elem = util_sparse_array_get(fl->arr, head_idx); in util_sparse_array_free_list_pop_elem()