| /external/tensorflow/tensorflow/core/framework/ |
| D | device_base.h | 133 struct CpuWorkerThreads { struct
139 void set_tensorflow_cpu_worker_threads(CpuWorkerThreads* t) { in set_tensorflow_cpu_worker_threads() argument
143 virtual const CpuWorkerThreads* tensorflow_cpu_worker_threads() const { in tensorflow_cpu_worker_threads()
289 CpuWorkerThreads* cpu_worker_threads_ = nullptr;
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | renamed_device.h | 46 const CpuWorkerThreads* tensorflow_cpu_worker_threads() const override { in tensorflow_cpu_worker_threads()
168 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 | 405 const DeviceBase::CpuWorkerThreads& worker_threads = in Initialize()
421 const DeviceBase::CpuWorkerThreads& worker_threads, const int num_examples, in CreateSparseFeatureRepresentation()
508 const DeviceBase::CpuWorkerThreads& worker_threads, const int num_examples, in CreateDenseFeatureRepresentation()
540 const DeviceBase::CpuWorkerThreads& worker_threads, const int num_examples, in ComputeSquaredNormPerExample()
|
| D | sparse_matmul_op.cc | 792 const DeviceBase::CpuWorkerThreads* thread_pool,
814 const DeviceBase::CpuWorkerThreads* thread_pool);
822 int num_cols, const DeviceBase::CpuWorkerThreads* thread_pool,
831 const DeviceBase::CpuWorkerThreads* thread_pool, MatrixR* buffer);
940 const DeviceBase::CpuWorkerThreads* thread_pool,
1150 const DeviceBase::CpuWorkerThreads* thread_pool) { in CreateSparseSlices()
1246 const DeviceBase::CpuWorkerThreads* thread_pool, MatrixR* buffer) { in ShuffleMatrix()
1316 const DeviceBase::CpuWorkerThreads* thread_pool, MatrixR* buffer, in CreateDenseSlices()
1372 void do_on_all_threads(const DeviceBase::CpuWorkerThreads* thread_pool, in do_on_all_threads()
1440 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 | 167 const DeviceBase::CpuWorkerThreads& worker_threads = in Compute()
|
| D | pooling_ops_common.h | 219 const DeviceBase::CpuWorkerThreads& worker_threads = in SpatialMaxPool()
484 const DeviceBase::CpuWorkerThreads& worker_threads =
658 const DeviceBase::CpuWorkerThreads& worker_threads =
|
| D | ctc_loss_op.cc | 209 DeviceBase::CpuWorkerThreads workers = in Compute()
|
| D | ctc_decoder_ops.cc | 258 const DeviceBase::CpuWorkerThreads& worker_threads = in Compute()
|
| D | scatter_functor.h | 176 const DeviceBase::CpuWorkerThreads& worker_threads =
|
| D | maxpooling_op.cc | 97 const DeviceBase::CpuWorkerThreads& worker_threads = in SpatialMaxPoolWithArgMaxHelper()
593 const DeviceBase::CpuWorkerThreads& worker_threads = in SpatialMaxPoolGradGrad()
1032 const DeviceBase::CpuWorkerThreads& worker_threads = in launch()
|
| D | avgpooling_op.cc | 394 const DeviceBase::CpuWorkerThreads& worker_threads = in Compute()
|
| D | pooling_ops_3d.cc | 592 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/transforms/utils/ |
| D | eval_utils.h | 54 tensorflow::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 | 342 const DeviceBase::CpuWorkerThreads& worker_threads = in operator ()()
559 const DeviceBase::CpuWorkerThreads& worker_threads = in operator ()()
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_qmatmul_op.cc | 456 const DeviceBase::CpuWorkerThreads& worker_threads = in GetBiasHandle()
|
| /external/tensorflow/tensorflow/core/runtime_fallback/kernel/ |
| D | kernel_fallback_execute_compat.cc | 777 const CpuWorkerThreads* tensorflow_cpu_worker_threads() const override { in tensorflow_cpu_worker_threads()
|