| /external/tensorflow/tensorflow/core/framework/ |
| D | device_base.h | 129 struct CpuWorkerThreads { struct
135 void set_tensorflow_cpu_worker_threads(CpuWorkerThreads* t) { in set_tensorflow_cpu_worker_threads() argument
139 virtual const CpuWorkerThreads* tensorflow_cpu_worker_threads() const { in tensorflow_cpu_worker_threads()
277 CpuWorkerThreads* cpu_worker_threads_ = nullptr;
|
| /external/tensorflow/tensorflow/core/kernels/boosted_trees/ |
| D | quantile_ops.cc | 204 const DeviceBase::CpuWorkerThreads& worker_threads = in Compute()
263 const DeviceBase::CpuWorkerThreads& worker_threads = in Compute()
330 const DeviceBase::CpuWorkerThreads& worker_threads = in Compute()
377 const DeviceBase::CpuWorkerThreads& worker_threads = in Compute()
429 const DeviceBase::CpuWorkerThreads& worker_threads = in Compute()
488 const DeviceBase::CpuWorkerThreads& worker_threads = in Compute()
565 const DeviceBase::CpuWorkerThreads& worker_threads = in Compute()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | renamed_device.h | 46 const CpuWorkerThreads* tensorflow_cpu_worker_threads() const override { in tensorflow_cpu_worker_threads()
167 DeviceBase::CpuWorkerThreads eigen_worker_threads_;
|
| D | single_threaded_cpu_device.cc | 90 DeviceBase::CpuWorkerThreads eigen_worker_threads_;
|
| D | local_device.cc | 109 DeviceBase::CpuWorkerThreads eigen_worker_threads_;
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sdca_internal.h | 352 const DeviceBase::CpuWorkerThreads& worker_threads, int num_examples,
363 const DeviceBase::CpuWorkerThreads& worker_threads, int num_examples,
371 const DeviceBase::CpuWorkerThreads& worker_threads, int num_examples,
|
| D | sdca_internal.cc | 371 const DeviceBase::CpuWorkerThreads& worker_threads = in Initialize()
387 const DeviceBase::CpuWorkerThreads& worker_threads, const int num_examples, in CreateSparseFeatureRepresentation()
467 const DeviceBase::CpuWorkerThreads& worker_threads, const int num_examples, in CreateDenseFeatureRepresentation()
499 const DeviceBase::CpuWorkerThreads& worker_threads, const int num_examples, in ComputeSquaredNormPerExample()
|
| D | sparse_matmul_op.cc | 796 const DeviceBase::CpuWorkerThreads* thread_pool,
818 const DeviceBase::CpuWorkerThreads* thread_pool);
826 int num_cols, const DeviceBase::CpuWorkerThreads* thread_pool,
835 const DeviceBase::CpuWorkerThreads* thread_pool, MatrixR* buffer);
944 const DeviceBase::CpuWorkerThreads* thread_pool,
1141 const DeviceBase::CpuWorkerThreads* thread_pool) { in CreateSparseSlices()
1237 const DeviceBase::CpuWorkerThreads* thread_pool, MatrixR* buffer) { in ShuffleMatrix()
1307 const DeviceBase::CpuWorkerThreads* thread_pool, MatrixR* buffer, in CreateDenseSlices()
1363 void do_on_all_threads(const DeviceBase::CpuWorkerThreads* thread_pool, in do_on_all_threads()
1431 bool transpose_left, const DeviceBase::CpuWorkerThreads* thread_pool, in Compute()
[all …]
|
| D | sdca_ops.cc | 225 const DeviceBase::CpuWorkerThreads& worker_threads = in DoCompute()
283 const DeviceBase::CpuWorkerThreads& worker_threads = in Compute()
|
| D | sobol_op.cc | 160 const DeviceBase::CpuWorkerThreads& worker_threads = in Compute()
|
| D | pooling_ops_common.h | 221 const DeviceBase::CpuWorkerThreads& worker_threads = in SpatialMaxPool()
482 const DeviceBase::CpuWorkerThreads& worker_threads =
652 const DeviceBase::CpuWorkerThreads& worker_threads =
|
| D | ctc_decoder_ops.cc | 248 const DeviceBase::CpuWorkerThreads& worker_threads = in Compute()
|
| D | ctc_loss_op.cc | 200 DeviceBase::CpuWorkerThreads workers = in Compute()
|
| D | xsmm_conv2d.cc | 285 const DeviceBase::CpuWorkerThreads* const worker_threads = in CallLibxsmmConvGeneric()
|
| D | scatter_functor.h | 175 const DeviceBase::CpuWorkerThreads& worker_threads =
|
| D | maxpooling_op.cc | 95 const DeviceBase::CpuWorkerThreads& worker_threads = in SpatialMaxPoolWithArgMaxHelper()
569 const DeviceBase::CpuWorkerThreads& worker_threads = in SpatialMaxPoolGradGrad()
993 const DeviceBase::CpuWorkerThreads& worker_threads = in launch()
|
| D | avgpooling_op.cc | 382 const DeviceBase::CpuWorkerThreads& worker_threads = in Compute()
|
| D | pooling_ops_3d.cc | 569 const DeviceBase::CpuWorkerThreads& worker_threads = in launch()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | evaluation_utils.h | 49 DeviceBase::CpuWorkerThreads eigen_worker_threads_;
|
| /external/tensorflow/tensorflow/core/util/ctc/ |
| D | ctc_loss_calculator.h | 74 DeviceBase::CpuWorkerThreads* workers = nullptr) const;
119 DeviceBase::CpuWorkerThreads* workers) const { in CalculateLoss()
|
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | adjust_hue_op.cc | 215 const DeviceBase::CpuWorkerThreads& worker_threads = in DoCompute()
|
| D | adjust_saturation_op.cc | 194 const DeviceBase::CpuWorkerThreads& worker_threads = in DoCompute()
|
| D | crop_and_resize_op.cc | 340 const DeviceBase::CpuWorkerThreads& worker_threads = in operator ()()
546 const DeviceBase::CpuWorkerThreads& worker_threads = in operator ()()
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_qmatmul_op.cc | 448 const DeviceBase::CpuWorkerThreads& worker_threads = in GetBiasHandle()
|