| /external/tensorflow/tensorflow/core/framework/ |
| D | device_base_test.cc | 38 auto d = dbase.eigen_cpu_device(); in TEST()
44 auto d = dbase.eigen_cpu_device(); in TEST()
50 auto d = dbase.eigen_cpu_device(); in TEST()
56 auto d = dbase.eigen_cpu_device(); in TEST()
|
| D | device_base.cc | 93 const Eigen::ThreadPoolDevice* DeviceBase::eigen_cpu_device() { in eigen_cpu_device() function in tensorflow::DeviceBase
|
| D | device_base.h | 211 virtual const Eigen::ThreadPoolDevice* eigen_cpu_device();
|
| D | op_kernel.h | 1109 const Eigen::ThreadPoolDevice& eigen_cpu_device() const { in eigen_cpu_device() function
1110 return *device()->eigen_cpu_device(); in eigen_cpu_device()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | renamed_device.h | 71 const Eigen::ThreadPoolDevice* eigen_cpu_device() override { in eigen_cpu_device() function
75 return Device::eigen_cpu_device(); in eigen_cpu_device()
77 return underlying_device_->eigen_cpu_device(); in eigen_cpu_device()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | quantized_activation_ops.cc | 60 output->flat<T>().device(context->eigen_cpu_device()) = in Compute()
108 output->flat<T>().device(context->eigen_cpu_device()) = in Compute()
|
| D | nn_ops_test.cc | 779 Eigen::ThreadPoolDevice eigen_cpu_device(threadpool.AsEigenThreadPool(), in BM_LRNFloat() local
781 device->set_eigen_cpu_device(&eigen_cpu_device); in BM_LRNFloat()
860 Eigen::ThreadPoolDevice eigen_cpu_device(threadpool.AsEigenThreadPool(), in BM_AvgPool() local
862 device->set_eigen_cpu_device(&eigen_cpu_device); in BM_AvgPool()
950 Eigen::ThreadPoolDevice eigen_cpu_device(threadpool.AsEigenThreadPool(), in BM_AvgPoolBk() local
952 device->set_eigen_cpu_device(&eigen_cpu_device); in BM_AvgPoolBk()
1055 Eigen::ThreadPoolDevice eigen_cpu_device(threadpool.AsEigenThreadPool(), in BM_MaxPool() local
1057 device->set_eigen_cpu_device(&eigen_cpu_device); in BM_MaxPool()
1241 Eigen::ThreadPoolDevice eigen_cpu_device(threadpool.AsEigenThreadPool(), in BM_ReluFloat() local
1243 device->set_eigen_cpu_device(&eigen_cpu_device); in BM_ReluFloat()
[all …]
|
| D | count_ops.cc | 123 nonnegative.device(context->eigen_cpu_device()) = in Compute()
249 nonnegative.device(context->eigen_cpu_device()) = in Compute()
360 nonnegative.device(context->eigen_cpu_device()) = in Compute()
|
| D | bincount_op.cc | 50 all_nonneg_t.scalar<bool>().device(context->eigen_cpu_device()) = in Compute()
79 output.device(context->eigen_cpu_device()) = in Compute()
95 all_nonneg_t.scalar<bool>().device(context->eigen_cpu_device()) = in Compute()
162 output.device(context->eigen_cpu_device()) = partial_bins.sum(reduce_dim); in Compute()
|
| D | quantized_batch_norm_op_test.cc | 220 Eigen::ThreadPoolDevice eigen_cpu_device(threadpool.AsEigenThreadPool(), 1); in TEST_F() local
227 eigen_cpu_device, const_input_float.tensor<float, 4>(), in TEST_F()
|
| D | pooling_ops_3d.cc | 267 tensor_in_slice.device(context->eigen_cpu_device()) = in launch()
272 tensor_out_slice.device(context->eigen_cpu_device()) = in launch()
277 out_backprop_slice.device(context->eigen_cpu_device()) = in launch()
292 .device(context->eigen_cpu_device()) += select_slice; in launch()
460 slices.device(context->eigen_cpu_device()) = in launch()
468 .device(context->eigen_cpu_device()) += slices.broadcast(bcast); in launch()
|
| D | sdca_ops.cc | 208 context->eigen_cpu_device(), example, in DoCompute()
271 prox_w.device(context->eigen_cpu_device()) = in Compute()
|
| D | sparse_reduce_op.cc | 138 s.device(ctx->eigen_cpu_device()) = v.sum(); in Run()
148 s.device(ctx->eigen_cpu_device()) = v.maximum(); in Run()
|
| D | lrn_op.cc | 116 out_shaped.device(context->eigen_cpu_device()) = in launch()
119 out_shaped.device(context->eigen_cpu_device()) = in_shaped * tmp.rsqrt(); in launch()
121 out_shaped.device(context->eigen_cpu_device()) = in launch()
|
| D | matmul_op_impl.h | 81 const Eigen::ThreadPoolDevice d = context->eigen_cpu_device(); in Conjugate()
101 const Eigen::ThreadPoolDevice d = context->eigen_cpu_device(); in Run()
136 const Eigen::ThreadPoolDevice d = context->eigen_cpu_device();
|
| D | set_kernels.cc | 279 out.device(ctx->eigen_cpu_device()) = out.constant(static_cast<int32>(0.0)); in Compute()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | evaluation_utils_test.cc | 33 dsimple.eigen_cpu_device()->getPool(); in TEST()
|
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | attention_ops.cc | 124 context->eigen_cpu_device()) = in Compute()
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_lrn_op.cc | 252 out_shaped.device(context->eigen_cpu_device()) = in MklDefaultToEigen()
255 out_shaped.device(context->eigen_cpu_device()) = in_shaped * tmp.rsqrt(); in MklDefaultToEigen()
257 out_shaped.device(context->eigen_cpu_device()) = in MklDefaultToEigen()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_compile_on_demand_op.cc | 92 run_options.set_intra_op_thread_pool(&ctx->eigen_cpu_device()); in Run()
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/ |
| D | tf_cpurt_kernels.cc | 106 const Eigen::ThreadPoolDevice* eigen = host_cpu->eigen_cpu_device(); in GetWorkerThreads()
|
| D | tf_jitrt_kernels.cc | 190 const Eigen::ThreadPoolDevice* eigen = host_cpu->eigen_cpu_device(); in GetWorkerThreads()
|
| /external/tensorflow/tensorflow/core/runtime_fallback/kernel/ |
| D | kernel_fallback_execute_compat.cc | 790 const Eigen::ThreadPoolDevice* eigen_cpu_device() override { in eigen_cpu_device() function in tensorflow::tfd::__anone1369dfd0711::DeviceWithCustomAllocator
791 return device_->eigen_cpu_device(); in eigen_cpu_device()
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | xrt_execute_op.cc | 316 run_options.set_intra_op_thread_pool(&context->eigen_cpu_device()); in RunExecutable()
|
| /external/tensorflow/tensorflow/compiler/jit/kernels/ |
| D | xla_ops.cc | 183 run_options.set_intra_op_thread_pool(&ctx->eigen_cpu_device()); in RunExecutable()
|