| /aosp_15_r20/external/executorch/extension/parallel/test/ |
| H A D | thread_parallel_test.cpp | 18 using ::executorch::extension::parallel_for;
51 EXPECT_TRUE(parallel_for(0, 10, 1, [this](int64_t begin, int64_t end) { in TEST_F()
61 EXPECT_TRUE(parallel_for(0, 10, 1, [this](int64_t begin, int64_t end) { in TEST_F()
75 EXPECT_FALSE(parallel_for(10, 0, 1, [this](int64_t begin, int64_t end) { in TEST_F()
87 EXPECT_FALSE(parallel_for(6, 5, 1, [this](int64_t begin, int64_t end) { in TEST_F()
98 EXPECT_TRUE(parallel_for(0, 5, 1, [this](int64_t begin, int64_t end) { in TEST_F()
111 EXPECT_TRUE(parallel_for(5, 10, 1, [this](int64_t begin, int64_t end) { in TEST_F()
124 EXPECT_TRUE(parallel_for(2, 8, 1, [this](int64_t begin, int64_t end) { in TEST_F()
140 EXPECT_TRUE(parallel_for(0, 10, 2, [this](int64_t begin, int64_t end) { in TEST_F()
150 EXPECT_TRUE(parallel_for(3, 8, 2, [this](int64_t begin, int64_t end) { in TEST_F()
[all …]
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/test/ |
| H A D | test_parallel.cpp | 50 at::parallel_for(0, 10, 1, [&](int64_t begin, int64_t end) { in TEST()
71 at::parallel_for(0, 10, 1, [&](int64_t begin, int64_t end) { in TEST()
72 at::parallel_for(0, 10, 1, [&](int64_t begin, int64_t end) { in TEST()
82 at::parallel_for(0, 10, 1, [&](int64_t begin, int64_t end) { in TEST()
96 at::parallel_for(0, 10, 1, [&](int64_t begin, int64_t end) { in TEST()
104 at::parallel_for(0, 1, 1000, [&](int64_t begin, int64_t end) { in TEST()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/parallel_for/ |
| H A D | __init__.py | 17 from tensorflow.python.ops.parallel_for import * # pylint: disable=wildcard-import
18 from tensorflow.python.ops.parallel_for.control_flow_ops import for_loop
19 from tensorflow.python.ops.parallel_for.control_flow_ops import pfor
20 from tensorflow.python.ops.parallel_for.gradients import batch_jacobian
21 from tensorflow.python.ops.parallel_for.gradients import jacobian
|
| H A D | index.md | 74 …w/blob/8b000ce0d5395d399e08791ae9589b41358f651d/tensorflow/python/ops/parallel_for/control_flow_op…
107 …low/blob/349172cf0ac29ba1346d244a40dc4761b4600f2e/tensorflow/python/ops/parallel_for/pfor.py#L2653)
135 tensorflow/python/ops/parallel_for/pfor.py (the user-facing APIs are defined in
152 …parallel_for/pfor.py#L3505-L3522) ([example](https://github.com/tensorflow/tensorflow/blob/8b202f0…
154 …w/blob/349172cf0ac29ba1346d244a40dc4761b4600f2e/tensorflow/python/ops/parallel_for/pfor.py#L3276-L…
156 …w/blob/349172cf0ac29ba1346d244a40dc4761b4600f2e/tensorflow/python/ops/parallel_for/pfor.py#L3360-L…
171 …parallel_for/pfor.py#L4499); [example](https://github.com/tensorflow/tensorflow/blob/8b202f08d52e8…
179 …w/blob/349172cf0ac29ba1346d244a40dc4761b4600f2e/tensorflow/python/ops/parallel_for/pfor.py#L5001) …
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | batch_norm_kernel.cpp | 99 at::parallel_for(0, n_batch * n_channel, 1, [&](int64_t begin, int64_t end) { in batch_norm_cpu_contiguous_impl()
151 at::parallel_for(0, n_batch * image_size, 1, [&](int64_t begin, int64_t end) { in batch_norm_cpu_channels_last_impl()
193 at::parallel_for(0, n_channel, 1, [&](int64_t begin, int64_t end) { in batch_norm_cpu_collect_stats_contiguous_impl()
253 at::parallel_for(0, N, 1, [&](int64_t begin, int64_t end) { in batch_norm_cpu_collect_stats_channels_last_impl()
269 at::parallel_for(0, n_channel, 1, [&](int64_t begin, int64_t end) { in batch_norm_cpu_collect_stats_channels_last_impl()
282 at::parallel_for(0, N, 1, [&](int64_t begin, int64_t end) { in batch_norm_cpu_collect_stats_channels_last_impl()
298 at::parallel_for(0, n_channel, 1, [&](int64_t begin, int64_t end) { in batch_norm_cpu_collect_stats_channels_last_impl()
332 …at::parallel_for(0, (n_channel + TILE_SIZE - 1) / TILE_SIZE, 1, [&](int64_t tile_idx_begin, int64_… in batch_norm_cpu_collect_stats_channels_last_impl()
356 …at::parallel_for(0, (n_channel + TILE_SIZE - 1) / TILE_SIZE, 1, [&](int64_t tile_idx_begin, int64_… in batch_norm_cpu_collect_stats_channels_last_impl()
378 at::parallel_for(0, n_channel, 1, [&](int64_t begin, int64_t end) { in batch_norm_cpu_collect_stats_channels_last_impl()
[all …]
|
| H A D | PaddingKernel.cpp | 182 at::parallel_for(0, channels * output_width, 1, [&](int64_t begin, int64_t end) { in cpu_padding()
194 at::parallel_for(0, channels * output_height, 1, [&](int64_t begin, int64_t end) { in cpu_padding()
209 … at::parallel_for(0, channels * output_depth * output_height, 1, [&](int64_t begin, int64_t end) { in cpu_padding()
268 at::parallel_for(0, nbatch * output_height * output_width, 1, [&](int64_t begin, int64_t end) { in cpu_padding_channels_last()
285 …at::parallel_for(0, nbatch * output_depth * output_height * output_width, 1, [&](int64_t begin, in… in cpu_padding_channels_last()
342 at::parallel_for(0, channels, 1, [&](int64_t begin, int64_t end) { in cpu_padding_backward()
352 at::parallel_for(0, channels, 1, [&](int64_t begin, int64_t end) { in cpu_padding_backward()
368 at::parallel_for(0, channels, 1, [&](int64_t begin, int64_t end) { in cpu_padding_backward()
430 at::parallel_for(0, nbatch, 1, [&](int64_t begin, int64_t end) { in cpu_padding_backward_channels_last()
447 at::parallel_for(0, nbatch, 1, [&](int64_t begin, int64_t end) { in cpu_padding_backward_channels_last()
|
| H A D | WeightNormKernel.cpp | 30 at::parallel_for(0, M, 1, [&](int64_t begin, int64_t end) { in weight_norm_first_dim_kernel()
140 at::parallel_for(0, M, 1, [&](int64_t begin, int64_t end) { in weight_norm_last_dim_kernel()
168 at::parallel_for(0, M, 1, [&](int64_t begin, int64_t end) { in weight_norm_last_dim_kernel()
192 at::parallel_for(0, M, 1, [&](int64_t begin, int64_t end) { in weight_norm_backward_first_dim_kernel()
339 at::parallel_for(0, M, 1, [&](int64_t begin, int64_t end) { in weight_norm_backward_last_dim_kernel()
377 at::parallel_for(0, M, 1, [&](int64_t begin, int64_t end) { in weight_norm_backward_last_dim_kernel()
|
| H A D | UpSampleMoreKernel.cpp | 209 at::parallel_for(0, channels, at::internal::GRAIN_SIZE / output_slice_size, loop1d); in cpu_upsample_nearest_backward()
212 at::parallel_for(0, channels, at::internal::GRAIN_SIZE / output_slice_size , loop2d); in cpu_upsample_nearest_backward()
216 at::parallel_for(0, channels, at::internal::GRAIN_SIZE / output_slice_size, loop3d); in cpu_upsample_nearest_backward()
325 at::parallel_for(0, num_batches, 0, loop2d); in cpu_upsample_nearest_backward_channels_last()
329 at::parallel_for(0, num_batches, 0, loop3d); in cpu_upsample_nearest_backward_channels_last()
579 at::parallel_for(0, channels, at::internal::GRAIN_SIZE / output_slice_size / 2, loop1d); in cpu_upsample_linear_backward()
582 at::parallel_for(0, channels, at::internal::GRAIN_SIZE / output_slice_size / 4, loop2d); in cpu_upsample_linear_backward()
586 at::parallel_for(0, channels, at::internal::GRAIN_SIZE / output_slice_size / 8, loop3d); in cpu_upsample_linear_backward()
732 at::parallel_for(0, num_batches, 0, loop2d); in cpu_upsample_linear_backward_channels_last()
736 at::parallel_for(0, num_batches, 0, loop3d); in cpu_upsample_linear_backward_channels_last()
|
| H A D | AdaptiveAvgPoolKernel.cpp | 37 at::parallel_for(0, channels, 0, [&](int64_t begin, int64_t end) { in cpu_adaptive_avg_pool2d()
92 at::parallel_for(0, nbatch * output_height * output_width, 0, [&](int64_t begin, int64_t end) { in cpu_adaptive_avg_pool2d_channels_last()
180 at::parallel_for(0, nbatch * output_height * output_width, 0, [&](int64_t begin, int64_t end) { in cpu_adaptive_avg_pool2d_channels_last()
275 at::parallel_for(0, channels, 0, [&](int64_t begin, int64_t end) { in cpu_adaptive_avg_pool2d_backward()
326 at::parallel_for(0, nbatch, 0, [&](int64_t begin, int64_t end) { in cpu_adaptive_avg_pool2d_backward_channels_last()
434 at::parallel_for(0, channels, 0, [&](int64_t begin, int64_t end) { in cpu_adaptive_avg_pool3d()
500 …at::parallel_for(0, nbatch * output_depth * output_height * output_width, 0, [&](int64_t begin, in… in cpu_adaptive_avg_pool3d_channels_last()
597 …at::parallel_for(0, nbatch * output_depth * output_height * output_width, 0, [&](int64_t begin, in… in cpu_adaptive_avg_pool3d_channels_last()
702 at::parallel_for(0, channels, 0, [&](int64_t begin, int64_t end) { in cpu_adaptive_avg_pool3d_backward()
762 at::parallel_for(0, nbatch, 0, [&](int64_t begin, int64_t end) { in cpu_adaptive_avg_pool3d_backward_channels_last()
|
| H A D | PixelShuffleKernel.cpp | 41 at::parallel_for(0, numel, 0, [&](int64_t begin, int64_t end) { in cpu_pixel_shuffle()
75 at::parallel_for(0, nbatch * height, 0, [&](int64_t begin, int64_t end) { in cpu_pixel_shuffle_channels_last()
140 at::parallel_for(0, numel, 0, [&](int64_t begin, int64_t end) { in cpu_pixel_unshuffle()
182 at::parallel_for(0, numel, 0, [&](int64_t begin, int64_t end) { in cpu_pixel_unshuffle_channels_last()
|
| H A D | AdaptiveMaxPoolKernel.cpp | 40 at::parallel_for(0, channels, 0, [&](int64_t begin, int64_t end) { in cpu_adaptive_max_pool2d()
118 at::parallel_for(0, nbatch * output_height * output_width, 0, [&](int64_t begin, int64_t end) { in cpu_adaptive_max_pool2d_channels_last()
233 at::parallel_for(0, nbatch * output_height * output_width, 0, [&](int64_t begin, int64_t end) { in cpu_adaptive_max_pool2d_channels_last()
363 at::parallel_for(0, channels, 0, [&](int64_t begin, int64_t end) { in cpu_adaptive_max_pool2d_backward()
411 at::parallel_for(0, nbatch, 0, [&](int64_t begin, int64_t end) { in cpu_adaptive_max_pool2d_backward_channels_last()
508 at::parallel_for(0, channels, 0, [&](int64_t begin, int64_t end) { in cpu_adaptive_max_pool3d()
594 …at::parallel_for(0, nbatch * output_depth * output_height * output_width, 0, [&](int64_t begin, in… in cpu_adaptive_max_pool3d_channels_last()
717 …at::parallel_for(0, nbatch * output_depth * output_height * output_width, 0, [&](int64_t begin, in… in cpu_adaptive_max_pool3d_channels_last()
855 at::parallel_for(0, channels, 0, [&](int64_t begin, int64_t end) { in cpu_adaptive_max_pool3d_backward()
907 at::parallel_for(0, nbatch, 0, [&](int64_t begin, int64_t end) { in cpu_adaptive_max_pool3d_backward_channels_last()
|
| H A D | SoftMaxKernel.cpp | 54 parallel_for(0, outer_size, 0, [&](int64_t begin, int64_t end) { in _vec_log_softmax_lastdim()
121 parallel_for(0, outer_size, 0, [&](int64_t begin, int64_t end) { in _vec_softmax_lastdim()
156 parallel_for(0, outer_size, 0, [&](int64_t begin, int64_t end) { in _vec_softmax_lastdim()
221 parallel_for( in _vec_host_softmax_backward_lastdim()
280 parallel_for( in _vec_softmax_backward()
365 parallel_for( in _vec_softmax_backward()
491 parallel_for( in _vec_log_softmax_backward()
575 parallel_for( in _vec_log_softmax_backward()
703 parallel_for( in _vec_softmax()
810 parallel_for( in _vec_softmax()
[all …]
|
| H A D | AvgPoolKernel.cpp | 43 at::parallel_for(0, numel, 0, [&](int64_t begin, int64_t end) { in cpu_avg_pool2d()
130 at::parallel_for(0, nbatch * output_height * output_width, 0, [&](int64_t begin, int64_t end) { in cpu_avg_pool2d_channels_last()
245 at::parallel_for(0, nbatch * output_height * output_width, 0, [&](int64_t begin, int64_t end) { in cpu_avg_pool2d_channels_last()
372 at::parallel_for(0, channels, 0, [&](int64_t begin, int64_t end) { in cpu_avg_pool2d_backward()
441 at::parallel_for(0, nbatch, 0, [&](int64_t begin, int64_t end) { in cpu_avg_pool2d_backward_channels_last()
578 at::parallel_for(0, numel, 0, [&](int64_t begin, int64_t end) { in cpu_avg_pool3d()
674 …at::parallel_for(0, nbatch * output_depth * output_height * output_width, 0, [&](int64_t begin, in… in cpu_avg_pool3d_channels_last()
798 …at::parallel_for(0, nbatch * output_depth * output_height * output_width, 0, [&](int64_t begin, in… in cpu_avg_pool3d_channels_last()
934 at::parallel_for(0, channels, 0, [&](int64_t begin, int64_t end) { in cpu_avg_pool3d_backward()
1012 at::parallel_for(0, nbatch, 0, [&](int64_t begin, int64_t end) { in cpu_avg_pool3d_backward_channels_last()
|
| H A D | MaxUnpoolKernel.cpp | 60 at::parallel_for(0, numel, 0, [&](int64_t begin, int64_t end) { in cpu_max_unpool()
124 at::parallel_for(0, nbatch * input_image_size, 0, [&](int64_t begin, int64_t end) { in cpu_max_unpool_channels_last()
197 at::parallel_for(0, numel, 0, [&](int64_t begin, int64_t end) { in cpu_max_unpool_backward()
|
| H A D | SpmmReduceKernel.cpp | 200 at::parallel_for(0, M, 1, [&](int64_t begin, int64_t end) { in spmm_reduce_arg_kernel_impl()
269 at::parallel_for(0, nnz, 1, [&](int64_t begin, int64_t end) { in spmm_reduce_backward_input_kernel_impl()
320 at::parallel_for(0, M, 1, [&](int64_t begin, int64_t end) { in spmm_reduce_backward_input_arg_kernel_impl()
364 at::parallel_for(0, nnz, 1, [&](int64_t begin, int64_t end) { in spmm_reduce_normalize_values_kernel_impl()
402 at::parallel_for(0, M, 1, [&](int64_t begin, int64_t end) { in spmm_reduce_backward_other_arg_kernel_impl()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| H A D | qembeddingbag_prepack.cpp | 100 at::parallel_for( in prepack()
123 at::parallel_for( in prepack()
260 at::parallel_for( in qembeddingbag_byte_prepack_out()
271 at::parallel_for( in qembeddingbag_byte_prepack_out()
406 at::parallel_for( in _qembeddingbag_nbit_prepack_helper()
418 at::parallel_for( in _qembeddingbag_nbit_prepack_helper()
|
| H A D | qembeddingbag_unpack.cpp | 82 at::parallel_for(0, input_rows, 1, [&](int32_t start_idx, int32_t end_idx) { in unpack()
137 at::parallel_for(0, input_rows, 1, [&](int64_t start_idx, int64_t end_idx) { in qembeddingbag_byte_unpack_out()
206 at::parallel_for(0, input_rows, 1, [&](int64_t start_idx, int64_t end_idx) { in _qembeddingbag_nbit_unpack_helper()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | AdaptiveAveragePooling3d.cpp | 41 at::parallel_for(0, sizeD, 1, [&](int64_t start, int64_t end) { in adaptive_avg_pool3d_out_frame()
156 at::parallel_for(0, n, 1, [&](int64_t start, int64_t end) { in adaptive_avg_pool3d_out_cpu_template()
189 at::parallel_for(0, sizeD, 1, [&](int64_t start, int64_t end) { in adaptive_avg_pool3d_backward_out_frame()
275 at::parallel_for(0, n, 1, [&](int64_t start, int64_t end) { in adaptive_avg_pool3d_backward_out_cpu_template()
|
| H A D | TensorIteratorReduce.cpp | 57 at::parallel_for(0, iter.numel(), internal::GRAIN_SIZE, [&](int64_t begin, int64_t end) { in two_pass_reduction()
111 at::parallel_for(0, cols, 1, [&](int64_t begin, int64_t end) { in parallel_dim_reduction()
159 at::parallel_for(0, cols, 1, [&](int64_t begin, int64_t end) { in foreach_reduced_elt()
|
| H A D | Unique.cpp | 58 at::parallel_for(0, numel, grain_size, [&](int64_t begin, int64_t end) { in unique_cpu_bool_template()
99 at::parallel_for(0, numel, grain_size, [&](int64_t begin, int64_t end) { in unique_cpu_bool_template()
197 at::parallel_for(0, numel, grain_size, [&](int64_t begin, int64_t end) { in unique_cpu_sorted_template()
234 at::parallel_for(0, numel, grain_size, [&](int64_t begin, int64_t end) { in unique_cpu_sorted_template()
257 at::parallel_for(0, unique_count, grain_size, [&](int64_t begin, int64_t end) { in unique_cpu_sorted_template()
|
| H A D | FractionalMaxPool2d.cpp | 139 at::parallel_for(0, numPlanes, 0, [&](int64_t start, int64_t end) { in fractional_max_pool2d_out_single_batch_frame()
210 at::parallel_for(0, numBatch, 0, [&](int64_t start, int64_t end) { in fractional_max_pool2d_out_frame()
230 at::parallel_for(0, numPlanes, 0, [&](int64_t start, int64_t end) { in fractional_max_pool2d_backward_out_single_batch_frame()
265 at::parallel_for(0, numBatch, 0, [&](int64_t start, int64_t end) { in fractional_max_pool2d_backward_out_frame()
|
| H A D | TriangularOps.cpp | 57 parallel_for(0, n, 0, [&](int64_t start, int64_t end) { in apply_triu_tril_single()
70 parallel_for(0, n, 0, [&](int64_t start, int64_t end) { in apply_triu_tril_single()
108 parallel_for(0, batchsize, 0, [&](int64_t start, int64_t end) { in apply_triu_tril()
|
| H A D | AdaptiveMaxPooling3d.cpp | 100 at::parallel_for(0, sizeD, 0, [&](int64_t start, int64_t end) { in adaptive_max_pool3d_single_out_frame()
179 at::parallel_for(0, sizeB, 0, [&](int64_t start, int64_t end) { in adaptive_max_pool3d_out_frame()
205 at::parallel_for(0, sizeD, 0, [&](int64_t start, int64_t end) { in adaptive_max_pool3d_backward_single_out_frame()
245 at::parallel_for(0, sizeB, 0, [&](int64_t start, int64_t end) { in adaptive_max_pool3d_backward_out_frame()
|
| H A D | FractionalMaxPool3d.cpp | 112 at::parallel_for(0, numPlanes, 0, [&](int64_t start, int64_t end) { in fractional_max_pool3d_out_single_batch_frame()
194 at::parallel_for(0, numBatch, 0, [&](int64_t start, int64_t end) { in fractional_max_pool3d_out_frame()
270 at::parallel_for(0, numPlanes, 0, [&](int64_t start, int64_t end) { in fractional_max_pool3d_backward_out_single_batch_frame()
311 at::parallel_for(0, numBatch, 0, [&](int64_t start, int64_t end) { in fractional_max_pool3d_backward_out_frame()
|
| /aosp_15_r20/external/executorch/extension/parallel/ |
| H A D | thread_parallel.h | 32 bool parallel_for(
50 using ::executorch::extension::parallel_for;
|