Home
last modified time | relevance | path

Searched refs:parallelFor (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Ddense_update_functor.cc43 d.parallelFor(update.data()->size(), in operator ()()
64 d.parallelFor( in operator ()()
Dredux_functor.h105 device.parallelFor(num_blocks, cost, compute); in operator()
Dtranspose_functor_cpu.cc65 device.parallelFor(in.NumElements(), cost, std::move(transpose_fn)); in TransposeSimple()
Dcwise_op_select.cc323 d.parallelFor(batch, cost, work); in operator ()()
Dunary_ops_composition.cc135 device.parallelFor(in.NumElements(), cost, AlignBlockSize, in Compute()
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorDeviceThreadPool.h182 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()
DTensorExecutor.h154 device.parallelFor(size, evaluator.costPerCoeff(Vectorizable),
/external/tensorflow/tensorflow/core/lib/core/
Dthreadpool.cc101 device.parallelFor( in ParallelFor()