Home
last modified time | relevance | path

Searched refs:CpuWorkerThreads (Results 1 – 23 of 23) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Ddevice_base.h126 struct CpuWorkerThreads { struct
132 void set_tensorflow_cpu_worker_threads(CpuWorkerThreads* t) { in set_tensorflow_cpu_worker_threads() argument
136 virtual const CpuWorkerThreads* tensorflow_cpu_worker_threads() const { in tensorflow_cpu_worker_threads()
266 CpuWorkerThreads* cpu_worker_threads_ = nullptr;
/external/tensorflow/tensorflow/core/kernels/boosted_trees/
Dquantile_ops.cc203 const DeviceBase::CpuWorkerThreads& worker_threads = in Compute()
264 const DeviceBase::CpuWorkerThreads& worker_threads = in Compute()
312 const DeviceBase::CpuWorkerThreads& worker_threads = in Compute()
365 const DeviceBase::CpuWorkerThreads& worker_threads = in Compute()
424 const DeviceBase::CpuWorkerThreads& worker_threads = in Compute()
501 const DeviceBase::CpuWorkerThreads& worker_threads = in Compute()
/external/tensorflow/tensorflow/core/kernels/
Dsdca_internal.h352 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,
Dsdca_internal.cc371 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()
Dsparse_matmul_op.cc797 const DeviceBase::CpuWorkerThreads* thread_pool,
819 const DeviceBase::CpuWorkerThreads* thread_pool);
827 int num_cols, const DeviceBase::CpuWorkerThreads* thread_pool,
836 const DeviceBase::CpuWorkerThreads* thread_pool, MatrixR* buffer);
945 const DeviceBase::CpuWorkerThreads* thread_pool,
1142 const DeviceBase::CpuWorkerThreads* thread_pool) { in CreateSparseSlices()
1238 const DeviceBase::CpuWorkerThreads* thread_pool, MatrixR* buffer) { in ShuffleMatrix()
1306 const DeviceBase::CpuWorkerThreads* thread_pool, MatrixR* buffer, in CreateDenseSlices()
1362 void do_on_all_threads(const DeviceBase::CpuWorkerThreads* thread_pool, in do_on_all_threads()
1429 bool transpose_left, const DeviceBase::CpuWorkerThreads* thread_pool, in Compute()
[all …]
Dsdca_ops.cc224 const DeviceBase::CpuWorkerThreads& worker_threads = in DoCompute()
282 const DeviceBase::CpuWorkerThreads& worker_threads = in Compute()
Dpooling_ops_common.h183 const DeviceBase::CpuWorkerThreads& worker_threads = in SpatialMaxPool()
430 const DeviceBase::CpuWorkerThreads& worker_threads =
600 const DeviceBase::CpuWorkerThreads& worker_threads =
Dctc_loss_op.cc162 DeviceBase::CpuWorkerThreads workers = in Compute()
Dadjust_saturation_op.cc192 const DeviceBase::CpuWorkerThreads& worker_threads = in DoCompute()
Dadjust_hue_op.cc215 const DeviceBase::CpuWorkerThreads& worker_threads = in DoCompute()
Dctc_decoder_ops.cc236 const DeviceBase::CpuWorkerThreads& worker_threads = in Compute()
Dxsmm_conv2d.cc228 const DeviceBase::CpuWorkerThreads* worker_threads = in CallLibxsmmConvGeneric()
Dmaxpooling_op.cc87 const DeviceBase::CpuWorkerThreads& worker_threads = in SpatialMaxPoolWithArgMaxHelper()
561 const DeviceBase::CpuWorkerThreads& worker_threads = in SpatialMaxPoolGradGrad()
961 const DeviceBase::CpuWorkerThreads& worker_threads = in launch()
Davgpooling_op.cc345 const DeviceBase::CpuWorkerThreads& worker_threads = in Compute()
Dcrop_and_resize_op.cc322 const DeviceBase::CpuWorkerThreads& worker_threads = in operator ()()
Dpooling_ops_3d.cc573 const DeviceBase::CpuWorkerThreads& worker_threads = in launch()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Devaluation_utils.h49 DeviceBase::CpuWorkerThreads eigen_worker_threads_;
/external/tensorflow/tensorflow/core/common_runtime/
Dsingle_threaded_cpu_device.cc81 DeviceBase::CpuWorkerThreads eigen_worker_threads_;
Drenamed_device.h50 const CpuWorkerThreads* tensorflow_cpu_worker_threads() const override { in tensorflow_cpu_worker_threads()
Dlocal_device.cc115 DeviceBase::CpuWorkerThreads eigen_worker_threads_;
/external/tensorflow/tensorflow/core/util/ctc/
Dctc_loss_calculator.h69 DeviceBase::CpuWorkerThreads* workers = nullptr) const;
113 DeviceBase::CpuWorkerThreads* workers) const { in CalculateLoss()
/external/tensorflow/tensorflow/contrib/image/kernels/
Dadjust_hsv_in_yiq_op.cc114 const DeviceBase::CpuWorkerThreads& worker_threads = in DoCompute()
/external/tensorflow/tensorflow/contrib/boosted_trees/kernels/
Dquantile_ops.cc448 const DeviceBase::CpuWorkerThreads& worker_threads = in Compute()
848 const DeviceBase::CpuWorkerThreads& worker_threads = in Compute()