Home
last modified time | relevance | path

Searched refs:eigen_cpu_device (Results 1 – 25 of 26) sorted by relevance

12

/external/tensorflow/tensorflow/core/framework/
Ddevice_base_test.cc39 auto d = dbase.eigen_cpu_device(); in TEST()
45 auto d = dbase.eigen_cpu_device(); in TEST()
51 auto d = dbase.eigen_cpu_device(); in TEST()
57 auto d = dbase.eigen_cpu_device(); in TEST()
Ddevice_base.cc48 const Eigen::ThreadPoolDevice* DeviceBase::eigen_cpu_device() { in eigen_cpu_device() function in tensorflow::DeviceBase
Ddevice_base.h206 virtual const Eigen::ThreadPoolDevice* eigen_cpu_device();
Dop_kernel.h1111 const Eigen::ThreadPoolDevice& eigen_cpu_device() const { in eigen_cpu_device() function
1112 return *device()->eigen_cpu_device(); in eigen_cpu_device()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Devaluation_utils_test.cc32 EXPECT_EQ(dsimple.eigen_cpu_device()->numThreads(), in TEST()
35 dsimple.eigen_cpu_device()->getPool(); in TEST()
/external/tensorflow/tensorflow/core/common_runtime/
Drenamed_device.h71 const Eigen::ThreadPoolDevice* eigen_cpu_device() override { in eigen_cpu_device() function
72 return underlying_->eigen_cpu_device(); in eigen_cpu_device()
/external/tensorflow/tensorflow/core/kernels/
Dbincount_op.cc47 all_nonneg_t.scalar<bool>().device(context->eigen_cpu_device()) = in Compute()
82 output.device(context->eigen_cpu_device()) = partial_bins.sum(reduce_dims); in Compute()
Dquantized_activation_ops.cc47 output->flat<T>().device(context->eigen_cpu_device()) = in Compute()
82 output->flat<T>().device(context->eigen_cpu_device()) = in Compute()
Dnn_ops_test.cc739 Eigen::ThreadPoolDevice eigen_cpu_device(&wrapper, num_threads); in BM_LRNFloat() local
740 device->set_eigen_cpu_device(&eigen_cpu_device); in BM_LRNFloat()
821 Eigen::ThreadPoolDevice eigen_cpu_device(&wrapper, num_threads); in BM_AvgPool() local
822 device->set_eigen_cpu_device(&eigen_cpu_device); in BM_AvgPool()
913 Eigen::ThreadPoolDevice eigen_cpu_device(&wrapper, num_threads); in BM_AvgPoolBk() local
914 device->set_eigen_cpu_device(&eigen_cpu_device); in BM_AvgPoolBk()
1017 Eigen::ThreadPoolDevice eigen_cpu_device(&wrapper, num_threads); in BM_MaxPool() local
1018 device->set_eigen_cpu_device(&eigen_cpu_device); in BM_MaxPool()
1197 Eigen::ThreadPoolDevice eigen_cpu_device(&wrapper, num_threads); in BM_ReluFloat() local
1198 device->set_eigen_cpu_device(&eigen_cpu_device); in BM_ReluFloat()
Dpooling_ops_3d.cc270 tensor_in_slice.device(context->eigen_cpu_device()) = in launch()
275 tensor_out_slice.device(context->eigen_cpu_device()) = in launch()
280 out_backprop_slice.device(context->eigen_cpu_device()) = in launch()
295 .device(context->eigen_cpu_device()) += select_slice; in launch()
441 slices.device(context->eigen_cpu_device()) = in launch()
449 .device(context->eigen_cpu_device()) += slices.broadcast(bcast); in launch()
Dsdca_ops.cc207 context->eigen_cpu_device(), example, in DoCompute()
270 prox_w.device(context->eigen_cpu_device()) = in Compute()
Dquantized_batch_norm_op_test.cc222 Eigen::ThreadPoolDevice eigen_cpu_device(&wrapper, 1); in TEST_F() local
229 eigen_cpu_device, const_input_float.tensor<float, 4>(), in TEST_F()
Dsparse_reduce_op.cc136 s.device(ctx->eigen_cpu_device()) = v.sum(); in Run()
146 s.device(ctx->eigen_cpu_device()) = v.maximum(); in Run()
Dlrn_op.cc108 out_shaped.device(context->eigen_cpu_device()) = in launch()
111 out_shaped.device(context->eigen_cpu_device()) = in_shaped * tmp.rsqrt(); in launch()
113 out_shaped.device(context->eigen_cpu_device()) = in launch()
Dattention_ops.cc122 context->eigen_cpu_device()) = in Compute()
Dbatch_matmul_op_impl.h71 const Eigen::ThreadPoolDevice d = context->eigen_cpu_device(); in Conjugate()
90 const Eigen::ThreadPoolDevice d = context->eigen_cpu_device(); in Run()
120 const Eigen::ThreadPoolDevice d = context->eigen_cpu_device();
Dmkl_lrn_op.cc257 out_shaped.device(context->eigen_cpu_device()) = in MklDefaultToEigen()
260 out_shaped.device(context->eigen_cpu_device()) = in_shaped * tmp.rsqrt(); in MklDefaultToEigen()
262 out_shaped.device(context->eigen_cpu_device()) = in MklDefaultToEigen()
Dset_kernels.cc268 out.device(ctx->eigen_cpu_device()) = out.constant(static_cast<int32>(0.0)); in Compute()
Dconv_grad_ops_3d.cc489 C.device(context->eigen_cpu_device()) = A.contract(B, contract_dims); in Compute()
985 C.device(context->eigen_cpu_device()) += A.contract(B, contract_dims); in Compute()
Dconv_grad_filter_ops.cc394 C.device(context->eigen_cpu_device()) += A.contract(B, contract_dims); in Compute()
Dconv_grad_input_ops.cc485 C.device(context->eigen_cpu_device()) = A.contract(B, contract_dims); in Compute()
/external/tensorflow/tensorflow/compiler/jit/kernels/
Dxla_ops.cc363 run_options.set_intra_op_thread_pool(&ctx->eigen_cpu_device()); in Compute()
537 run_options.set_intra_op_thread_pool(&ctx->eigen_cpu_device()); in Compute()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_compile_on_demand_op.cc78 run_options.set_intra_op_thread_pool(&ctx->eigen_cpu_device()); in Run()
/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dxrt_execute_op.cc210 run_options.set_intra_op_thread_pool(&context->eigen_cpu_device()); in DoWork()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dmap_and_batch_dataset_op.cc139 std::move(captured_func), &ctx->eigen_cpu_device(), in MakeDataset()

12