| /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 | 88 const Eigen::ThreadPoolDevice* DeviceBase::eigen_cpu_device() { in eigen_cpu_device() function in tensorflow::DeviceBase
|
| D | device_base.h | 207 virtual const Eigen::ThreadPoolDevice* eigen_cpu_device();
|
| D | op_kernel.h | 1153 const Eigen::ThreadPoolDevice& eigen_cpu_device() const { in eigen_cpu_device() function
1154 return *device()->eigen_cpu_device(); in eigen_cpu_device()
|
| D | op_kernel.cc | 1712 return eigen_cpu_device(); in eigen_device()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | renamed_device.h | 70 const Eigen::ThreadPoolDevice* eigen_cpu_device() override { in eigen_cpu_device() function
74 return Device::eigen_cpu_device(); in eigen_cpu_device()
76 return underlying_device_->eigen_cpu_device(); in eigen_cpu_device()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | quantized_activation_ops.cc | 47 output->flat<T>().device(context->eigen_cpu_device()) = in Compute()
82 output->flat<T>().device(context->eigen_cpu_device()) = in Compute()
|
| D | bincount_op.cc | 48 all_nonneg_t.scalar<bool>().device(context->eigen_cpu_device()) = in Compute()
77 output.device(context->eigen_cpu_device()) = in Compute()
93 all_nonneg_t.scalar<bool>().device(context->eigen_cpu_device()) = in Compute()
128 output.device(context->eigen_cpu_device()) = partial_bins.sum(reduce_dim); in Compute()
|
| D | nn_ops_test.cc | 764 Eigen::ThreadPoolDevice eigen_cpu_device(threadpool.AsEigenThreadPool(), in BM_LRNFloat() local
766 device->set_eigen_cpu_device(&eigen_cpu_device); in BM_LRNFloat()
845 Eigen::ThreadPoolDevice eigen_cpu_device(threadpool.AsEigenThreadPool(), in BM_AvgPool() local
847 device->set_eigen_cpu_device(&eigen_cpu_device); in BM_AvgPool()
935 Eigen::ThreadPoolDevice eigen_cpu_device(threadpool.AsEigenThreadPool(), in BM_AvgPoolBk() local
937 device->set_eigen_cpu_device(&eigen_cpu_device); in BM_AvgPoolBk()
1040 Eigen::ThreadPoolDevice eigen_cpu_device(threadpool.AsEigenThreadPool(), in BM_MaxPool() local
1042 device->set_eigen_cpu_device(&eigen_cpu_device); in BM_MaxPool()
1219 Eigen::ThreadPoolDevice eigen_cpu_device(threadpool.AsEigenThreadPool(), in BM_ReluFloat() local
1221 device->set_eigen_cpu_device(&eigen_cpu_device); in BM_ReluFloat()
[all …]
|
| 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 | sdca_ops.cc | 208 context->eigen_cpu_device(), example, in DoCompute()
271 prox_w.device(context->eigen_cpu_device()) = in Compute()
|
| D | pooling_ops_3d.cc | 266 tensor_in_slice.device(context->eigen_cpu_device()) = in launch()
271 tensor_out_slice.device(context->eigen_cpu_device()) = in launch()
276 out_backprop_slice.device(context->eigen_cpu_device()) = in launch()
291 .device(context->eigen_cpu_device()) += select_slice; in launch()
437 slices.device(context->eigen_cpu_device()) = in launch()
445 .device(context->eigen_cpu_device()) += slices.broadcast(bcast); in launch()
|
| D | sparse_reduce_op.cc | 136 s.device(ctx->eigen_cpu_device()) = v.sum(); in Run()
146 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 | 72 const Eigen::ThreadPoolDevice d = context->eigen_cpu_device(); in Conjugate()
92 const Eigen::ThreadPoolDevice d = context->eigen_cpu_device(); in Run()
127 const Eigen::ThreadPoolDevice d = context->eigen_cpu_device();
|
| D | set_kernels.cc | 268 out.device(ctx->eigen_cpu_device()) = out.constant(static_cast<int32>(0.0)); in Compute()
|
| D | conv_grad_input_ops.h | 703 C.device(context->eigen_cpu_device()) = A.contract(B, contract_dims);
|
| D | conv_grad_ops_3d.cc | 500 C.device(context->eigen_cpu_device()) = A.contract(B, contract_dims); in Compute()
996 C.device(context->eigen_cpu_device()) += A.contract(B, contract_dims); in Compute()
|
| D | conv_grad_filter_ops.cc | 593 C.device(context->eigen_cpu_device()) += A.contract(B, contract_dims); 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 | 123 context->eigen_cpu_device()) = in Compute()
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_lrn_op.cc | 251 out_shaped.device(context->eigen_cpu_device()) = in MklDefaultToEigen()
254 out_shaped.device(context->eigen_cpu_device()) = in_shaped * tmp.rsqrt(); in MklDefaultToEigen()
256 out_shaped.device(context->eigen_cpu_device()) = in MklDefaultToEigen()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_compile_on_demand_op.cc | 84 run_options.set_intra_op_thread_pool(&ctx->eigen_cpu_device()); in Run()
|
| /external/tensorflow/tensorflow/compiler/jit/kernels/ |
| D | xla_ops.cc | 274 run_options.set_intra_op_thread_pool(&ctx->eigen_cpu_device()); in Compute()
510 run_options.set_intra_op_thread_pool(&ctx->eigen_cpu_device()); in Compute()
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | xrt_execute_op.cc | 314 run_options.set_intra_op_thread_pool(&context->eigen_cpu_device()); in RunExecutable()
|