Searched refs:parallelFor (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | dense_update_functor.cc | 43 d.parallelFor(update.data()->size(), in operator ()()
64 d.parallelFor( in operator ()()
|
| D | one_hot_op.h | 104 d.parallelFor(prefix_size, cost, func);
116 d.parallelFor(prefix_size * suffix_size, cost * suffix_size, func);
|
| D | redux_functor.h | 121 device.parallelFor(num_blocks, cost, compute); in operator()
184 device.parallelFor(num_blocks, cost, compute); in operator()
319 device.parallelFor(num_blocks, cost, compute); in operator()
|
| D | transpose_functor_cpu.cc | 65 device.parallelFor(in.NumElements(), cost, std::move(transpose_fn)); in TransposeSimple()
|
| D | gather_nd_op_cpu_impl.h | 118 d.parallelFor(batch_size, cost, compute_shard);
|
| D | cwise_op_select.cc | 395 d.parallelFor(batch, cost, work); in operator ()()
|
| D | unary_ops_composition.cc | 135 device.parallelFor(in.NumElements(), cost, AlignBlockSize, in Compute()
|
| D | training_ops.cc | 211 d.parallelFor(N, cost, shard); in operator ()()
238 d.parallelFor(N, cost, shard); in operator ()()
815 d.parallelFor(length, cost, shard); in operator ()()
|
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | TensorDeviceThreadPool.h | 182 void parallelFor(Index n, const TensorOpCost& cost, in parallelFor() function
266 void parallelFor(Index n, const TensorOpCost& cost, in parallelFor() function
268 parallelFor(n, cost, nullptr, std::move(f)); in parallelFor()
|
| D | TensorExecutor.h | 154 device.parallelFor(size, evaluator.costPerCoeff(Vectorizable),
|
| /external/tensorflow/tensorflow/core/platform/ |
| D | threadpool.cc | 207 threadpool_device_->parallelFor( in ParallelFor()
218 threadpool_device_->parallelFor(total, in ParallelForWithWorkerId()
|
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | non_max_suppression_op.cc | 495 d.parallelFor(length, cost, shard_nms); in BatchedNonMaxSuppressionOp()
532 d.parallelFor(length, cost_result, shard_result); in BatchedNonMaxSuppressionOp()
571 d.parallelFor(length, cost_copy_result, shard_copy_result); in BatchedNonMaxSuppressionOp()
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_quantize_op.cc | 394 d.parallelFor(input.NumElements(), cost, ParallelSub); in Compute()
|