Home
last modified time | relevance | path

Searched refs:write_index (Results 1 – 25 of 34) sorted by relevance

12

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/
Dcumsum_impl.cu23 …for (size_t write_index = blockIdx.x * blockDim.x + threadIdx.x; write_index < size; write_index +… in Copy() local
24 input[write_index] = output[write_index]; in Copy()
34 …for (size_t write_index = blockIdx.x * blockDim.x + threadIdx.x; write_index < num; write_index +=… in LeftMoveSum() local
35 i = write_index / dim2 % dim0; in LeftMoveSum()
36 k = write_index % dim2; in LeftMoveSum()
56 …for (size_t write_index = blockIdx.x * blockDim.x + threadIdx.x; write_index < num; write_index +=… in RightMoveSum() local
57 i = write_index / dim2 % dim0; in RightMoveSum()
58 k = write_index % dim2; in RightMoveSum()
77 …for (size_t write_index = blockIdx.x * blockDim.x + threadIdx.x; write_index < num; write_index +=… in CumSumKernelReverse() local
78 i = write_index / dim2 % dim0; in CumSumKernelReverse()
[all …]
Dcumprod_impl.cu23 …for (size_t write_index = blockIdx.x * blockDim.x + threadIdx.x; write_index < size; write_index +… in Copy() local
24 input[write_index] = output[write_index]; in Copy()
34 …for (size_t write_index = blockIdx.x * blockDim.x + threadIdx.x; write_index < num; write_index +=… in LeftMoveProd() local
35 i = write_index / dim2 % dim0; in LeftMoveProd()
36 k = write_index % dim2; in LeftMoveProd()
56 …for (size_t write_index = blockIdx.x * blockDim.x + threadIdx.x; write_index < num; write_index +=… in RightMoveProd() local
57 i = write_index / dim2 % dim0; in RightMoveProd()
58 k = write_index % dim2; in RightMoveProd()
77 …for (size_t write_index = blockIdx.x * blockDim.x + threadIdx.x; write_index < num; write_index +=… in CumProdKernelReverse() local
78 i = write_index / dim2 % dim0; in CumProdKernelReverse()
[all …]
Dscatter_nd_functor_impl.cu26 size_t write_index = 0; in ScatterNdUpdate() local
35 write_index += indices_i * out_strides[k] * unit_size; in ScatterNdUpdate()
38 write_index += j; in ScatterNdUpdate()
41 input[write_index] = updates[read_index]; in ScatterNdUpdate()
52 size_t write_index = 0; in ScatterNdAdd() local
61 write_index += indices_i * out_strides[k] * unit_size; in ScatterNdAdd()
64 write_index += j; in ScatterNdAdd()
67 MsAtomicAdd(&input[write_index], updates[read_index]); in ScatterNdAdd()
78 size_t write_index = 0; in ScatterNdSub() local
87 write_index += indices_i * out_strides[k] * unit_size; in ScatterNdSub()
[all …]
Dgatherv2.cu25 for (size_t write_index = blockIdx.x * blockDim.x + threadIdx.x; write_index < num; in GatherV2Kernel() local
26 write_index += blockDim.x * gridDim.x) { in GatherV2Kernel()
27 i = write_index / (output_dim1 * output_dim2) % output_dim0; in GatherV2Kernel()
28 j = write_index / output_dim2 % output_dim1; in GatherV2Kernel()
29 k = write_index % output_dim2; in GatherV2Kernel()
33 output[write_index] = input[read_index]; in GatherV2Kernel()
35 output[write_index] = 0; in GatherV2Kernel()
Dgathernd.cu24 for (int write_index = blockIdx.x * blockDim.x + threadIdx.x; write_index < num; in GatherNdKernel() local
25 write_index += blockDim.x * gridDim.x) { in GatherNdKernel()
26 i = write_index / output_dim1 % output_dim0; in GatherNdKernel()
27 j = write_index % output_dim1; in GatherNdKernel()
41 output[write_index] = input[read_index]; in GatherNdKernel()
43 output[write_index] = 0; in GatherNdKernel()
Dtensor_scatter_add.cu28 size_t write_index = 0; in TensorScatterAddKernel() local
37 write_index += indices_i * indices_stride[k]; in TensorScatterAddKernel()
40 write_index += j; in TensorScatterAddKernel()
41 out_bound |= write_index >= output_size; in TensorScatterAddKernel()
44 MsAtomicAdd(&output[write_index], update[read_index]); in TensorScatterAddKernel()
Dtensor_scatter_update.cu28 size_t write_index = 0; in TensorScatterUpdateKernel() local
37 write_index += indices_i * indices_stride[k]; in TensorScatterUpdateKernel()
40 write_index += j; in TensorScatterUpdateKernel()
41 out_bound |= write_index >= output_size; in TensorScatterUpdateKernel()
44 output[write_index] = update[read_index]; in TensorScatterUpdateKernel()
Dtensor_scatter_min.cu28 size_t write_index = 0; in TensorScatterMinKernel() local
37 write_index += indices_i * indices_stride[k]; in TensorScatterMinKernel()
40 write_index += j; in TensorScatterMinKernel()
41 out_bound |= write_index >= output_size; in TensorScatterMinKernel()
44 MsAtomicMin(&output[write_index], update[read_index]); in TensorScatterMinKernel()
Dtensor_scatter_max.cu28 size_t write_index = 0; in TensorScatterMaxKernel() local
37 write_index += indices_i * indices_stride[k]; in TensorScatterMaxKernel()
40 write_index += j; in TensorScatterMaxKernel()
41 out_bound |= write_index >= output_size; in TensorScatterMaxKernel()
44 MsAtomicMax(&output[write_index], update[read_index]); in TensorScatterMaxKernel()
Dtensor_scatter_sub.cu28 size_t write_index = 0; in TensorScatterSubKernel() local
37 write_index += indices_i * indices_stride[k]; in TensorScatterSubKernel()
40 write_index += j; in TensorScatterSubKernel()
41 out_bound |= write_index >= output_size; in TensorScatterSubKernel()
44 MsAtomicSub(&output[write_index], update[read_index]); in TensorScatterSubKernel()
Dscatter_nd.cu28 size_t write_index = 0; in ScatterNdKernel() local
37 write_index += indices_i * indices_stride[k]; in ScatterNdKernel()
40 write_index += j; in ScatterNdKernel()
41 out_bound |= write_index >= output_size; in ScatterNdKernel()
44 MsAtomicAdd(&output[write_index], update[read_index]); in ScatterNdKernel()
/third_party/pulseaudio/src/pulsecore/
Dmemblockq.c52 int64_t read_index, write_index; member
82 bq->read_index = bq->write_index = idx; in pa_memblockq_new()
163 …_UNLIKELY(bq->current_write->index + (int64_t) bq->current_write->chunk.length <= bq->write_index)) in fix_current_write()
171 …e (PA_LIKELY(bq->current_write != NULL) && PA_UNLIKELY(bq->current_write->index > bq->write_index)) in fix_current_write()
228 if (bq->read_index > bq->write_index) { in can_push()
229 int64_t d = bq->read_index - bq->write_index; in can_push()
237 …->blocks_tail ? bq->blocks_tail->index + (int64_t) bq->blocks_tail->chunk.length : bq->write_index; in can_push()
240 if (bq->write_index + (int64_t) l > end) in can_push()
241 if (bq->write_index + (int64_t) l - bq->read_index > (int64_t) bq->maxlength) in can_push()
252 delta = bq->write_index - old_write_index; in write_index_changed()
[all …]
/third_party/boost/boost/lockfree/
Dspsc_queue.hpp74 static size_t read_available(size_t write_index, size_t read_index, size_t max_size) in read_available() argument
76 if (write_index >= read_index) in read_available()
77 return write_index - read_index; in read_available()
79 const size_t ret = write_index + max_size - read_index; in read_available()
83 static size_t write_available(size_t write_index, size_t read_index, size_t max_size) in write_available() argument
85 size_t ret = read_index - write_index - 1; in write_available()
86 if (write_index >= read_index) in write_available()
93 size_t write_index = write_index_.load(memory_order_acquire); in read_available() local
95 return read_available(write_index, read_index, max_size); in read_available()
100 size_t write_index = write_index_.load(memory_order_relaxed); in write_available() local
[all …]
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
Dgathernd_cpu_kernel.cc91 for (size_t write_index = 0; write_index < num; write_index++) { in Launch() local
92 size_t i = write_index / output_dim1 % output_dim0; in Launch()
93 size_t j = write_index % output_dim1; in Launch()
102 output_addr[write_index] = input_addr[read_index]; in Launch()
/third_party/skia/third_party/externals/harfbuzz/src/
Dhb-bit-set.hh265 unsigned int write_index = 0; in del_pages() local
270 page_map[write_index++] = page_map[i]; in del_pages()
272 compact (compact_workspace, write_index); in del_pages()
273 resize (write_index); in del_pages()
440 unsigned int write_index = 0; in compact_pages() local
445 if (write_index < i) in compact_pages()
446 pages[write_index] = pages[i]; in compact_pages()
448 page_map[old_index_to_page_map_index[i]].index = write_index; in compact_pages()
449 write_index++; in compact_pages()
470 unsigned int write_index = 0; in process() local
[all …]
/third_party/harfbuzz/src/
Dhb-set.hh397 unsigned int write_index = 0; in del_pages() local
402 page_map[write_index++] = page_map[i]; in del_pages()
404 compact (compact_workspace, write_index); in del_pages()
405 resize (write_index); in del_pages()
553 unsigned int write_index = 0; in compact_pages() local
558 if (write_index < i) in compact_pages()
559 pages[write_index] = pages[i]; in compact_pages()
561 page_map[old_index_to_page_map_index[i]].index = write_index; in compact_pages()
562 write_index++; in compact_pages()
582 unsigned int write_index = 0; in process() local
[all …]
/third_party/pulseaudio/src/modules/
Dmodule-esound-sink.c95 size_t write_length, write_index; member
353 pa_assert(u->write_index < u->write_length); in do_write()
355 …_iochannel_write(u->io, (uint8_t*) u->write_data + u->write_index, u->write_length - u->write_inde… in do_write()
360 u->write_index += (size_t) r; in do_write()
361 pa_assert(u->write_index <= u->write_length); in do_write()
363 if (u->write_index == u->write_length) { in do_write()
366 u->write_index = u->write_length = 0; in do_write()
420 u->write_index = 0; in handle_response()
449 u->write_index = 0; in handle_response()
591 u->read_index = u->write_index = u->read_length = u->write_length = 0; in pa__init()
/third_party/pulseaudio/src/modules/bluetooth/
Dmodule-bluez5-device.c130 uint64_t write_index; member
310 u->write_index += (uint64_t) memchunk.length; in sco_process_render()
450 u->write_index += (uint64_t) u->write_memchunk.length; in a2dp_write_buffer()
490 u->write_index += (uint64_t) u->write_memchunk.length; in a2dp_write_buffer()
524 …length = u->a2dp_codec->encode_buffer(u->encoder_info, u->write_index / pa_frame_size(&u->encoder_… in a2dp_process_render()
841 u->read_index = u->write_index = 0; in setup_stream()
1067 … wi = pa_bytes_to_usec(u->write_index + u->write_block_size, &u->encoder_sample_spec); in sink_process_msg()
1070 wi = pa_bytes_to_usec(u->write_index, &u->encoder_sample_spec); in sink_process_msg()
1342 if (u->write_index <= 0) in write_block()
1412 if (have_sink && u->write_index == 0 && u->read_index <= 0) in thread_func()
[all …]
/third_party/libuv/src/unix/
Dstream.c709 size = uv__count_bufs(req->bufs + req->write_index, in uv__write_req_size()
710 req->nbufs - req->write_index); in uv__write_req_size()
732 buf = req->bufs + req->write_index; in uv__write_req_update()
742 req->write_index = buf - req->bufs; in uv__write_req_update()
744 return req->write_index == req->nbufs; in uv__write_req_update()
901 &(req->bufs[req->write_index]), in uv__write()
902 req->nbufs - req->write_index, in uv__write()
1478 req->write_index = 0; in uv_write2()
/third_party/ffmpeg/libavformat/
Dnut.h110 int write_index; member
Dnutenc.c592 static int write_index(NUTContext *nut, AVIOContext *bc) { in write_index() function
1017 if (!nut->write_index && 2*index > st->nb_index_entries) { in nut_write_packet()
1040 if (nut->write_index) { in nut_write_packet()
1189 av_assert1(nut->write_index); // sp_count should be 0 if no index is going to be written in nut_write_trailer()
1190 write_index(nut, dyn_bc); in nut_write_trailer()
1220 …{ "write_index", "Write index", OFFSET(write_index), AV_OPT_TYPE_BOO…
Dwtvenc.c183 static void write_index(AVFormatContext *s) in write_index() function
213 write_index(s); in finish_chunk()
437 write_index(s); in write_header()
/third_party/mesa3d/src/compiler/nir/
Dnir_opt_find_array_copies.c421 unsigned write_index, unsigned read_index, in handle_write() argument
478 dst_node->last_successful_write = write_index; in handle_write()
/third_party/libuv/include/uv/
Dunix.h262 unsigned int write_index; \
/third_party/pulseaudio/src/pulse/
Ddef.h719 int64_t write_index; member

12