| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | dense_update_functor.cc | 43 d.parallelFor(update.data()->size(), in operator ()()
64 d.parallelFor( in operator ()()
|
| H A D | one_hot_op.h | 104 d.parallelFor(prefix_size, cost, func);
116 d.parallelFor(prefix_size * suffix_size, cost * suffix_size, func);
|
| H A D | redux_functor.h | 121 device.parallelFor(num_blocks, cost, compute); in operator()
184 device.parallelFor(num_blocks, cost, compute); in operator()
314 device.parallelFor(num_blocks, cost, compute); in operator()
|
| H A D | transpose_functor_cpu.cc | 68 device.parallelFor(in.NumElements(), cost, std::move(transpose_fn)); in TransposeSimple()
|
| H A D | gather_nd_op_cpu_impl.h | 118 d.parallelFor(batch_size, cost, compute_shard);
|
| H A D | cwise_op_select.cc | 411 d.parallelFor(batch, cost, work); in operator ()()
|
| H A D | unary_ops_composition.cc | 135 device.parallelFor(in.NumElements(), cost, AlignBlockSize, in Compute()
|
| H A D | segment_reduction_ops_impl.h | 427 cpu_device.parallelFor(num_segments, cost, reductionWorker);
|
| H A D | training_ops.cc | 211 d.parallelFor(N, cost, shard); in operator ()()
238 d.parallelFor(N, cost, shard); in operator ()()
843 d.parallelFor(length, cost, shard); in operator ()()
|
| /aosp_15_r20/external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| H A D | TensorDeviceThreadPool.h | 190 void parallelFor(Index n, const TensorOpCost& cost, in parallelFor() function
237 void parallelFor(Index n, const TensorOpCost& cost, in parallelFor() function
239 parallelFor(n, cost, nullptr, std::move(f)); in parallelFor()
|
| H A D | TensorScan.h | 229 self.device().parallelFor(
246 self.device().parallelFor(
267 self.device().parallelFor(
303 self.device().parallelFor(
|
| H A D | TensorExecutor.h | 338 device.parallelFor(size, evaluator.costPerCoeff(Vectorizable),
395 device.parallelFor(tiling.block_mapper.blockCount(), tiling.cost,
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/platform/ |
| H A D | threadpool.cc | 207 threadpool_device_->parallelFor( in ParallelFor()
218 threadpool_device_->parallelFor(total, in ParallelForWithWorkerId()
|
| /aosp_15_r20/prebuilts/clang/host/linux-x86/clang-r530567/include/llvm/Support/ |
| D | Parallel.h | 230 void parallelFor(size_t Begin, size_t End, function_ref<void(size_t)> Fn);
234 parallelFor(0, End - Begin, [&](size_t I) { Fn(Begin[I]); }); in parallelForEach()
|
| /aosp_15_r20/prebuilts/clang/host/linux-x86/clang-r530567b/include/llvm/Support/ |
| D | Parallel.h | 230 void parallelFor(size_t Begin, size_t End, function_ref<void(size_t)> Fn);
234 parallelFor(0, End - Begin, [&](size_t I) { Fn(Begin[I]); }); in parallelForEach()
|
| /aosp_15_r20/prebuilts/clang/host/linux-x86/clang-r522817/include/llvm/Support/ |
| D | Parallel.h | 230 void parallelFor(size_t Begin, size_t End, function_ref<void(size_t)> Fn);
234 parallelFor(0, End - Begin, [&](size_t I) { Fn(Begin[I]); }); in parallelForEach()
|
| /aosp_15_r20/prebuilts/clang/host/linux-x86/clang-r536225/include/llvm/Support/ |
| D | Parallel.h | 230 void parallelFor(size_t Begin, size_t End, function_ref<void(size_t)> Fn);
234 parallelFor(0, End - Begin, [&](size_t I) { Fn(Begin[I]); }); in parallelForEach()
|
| /aosp_15_r20/external/swiftshader/third_party/llvm-16.0/llvm/include/llvm/Support/ |
| H A D | Parallel.h | 230 void parallelFor(size_t Begin, size_t End, function_ref<void(size_t)> Fn);
234 parallelFor(0, End - Begin, [&](size_t I) { Fn(Begin[I]); }); in parallelForEach()
|
| /aosp_15_r20/external/swiftshader/third_party/llvm-16.0/llvm/lib/DebugInfo/PDB/Native/ |
| H A D | GSIStreamBuilder.cpp | 200 parallelFor(0, Records.size(), [&](size_t I) { in finalizeBuckets()
235 parallelFor(0, IPHR_HASH, [&](size_t I) { in finalizeBuckets()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/image/ |
| H A D | non_max_suppression_op.cc | 518 d.parallelFor(length, cost, shard_nms); in BatchedNonMaxSuppressionOp()
555 d.parallelFor(length, cost_result, shard_result); in BatchedNonMaxSuppressionOp()
594 d.parallelFor(length, cost_copy_result, shard_copy_result); in BatchedNonMaxSuppressionOp()
|
| H A D | resize_nearest_neighbor_op.cc | 206 d.parallelFor(N, cost, ParallelResize); in operator ()()
|
| /aosp_15_r20/external/swiftshader/third_party/llvm-16.0/llvm/lib/Support/ |
| H A D | Parallel.cpp | 195 void llvm::parallelFor(size_t Begin, size_t End, in parallelFor() function in llvm
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/ |
| H A D | mkl_quantize_op.cc | 403 d.parallelFor(input.NumElements(), cost, ParallelSub); in Compute()
|