Home
last modified time | relevance | path

Searched defs:CpuDevice (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dslice_op_cpu_impl.h27 using CpuDevice = Eigen::ThreadPoolDevice; variable
/external/tensorflow/tensorflow/core/kernels/image/
Dmirror_pad_op_cpu_impl.h27 using CpuDevice = Eigen::ThreadPoolDevice; variable
Dmirror_pad_op.cc149 using CpuDevice = Eigen::ThreadPoolDevice; typedef
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dcpu_device.cc27 CpuDevice::CpuDevice(int id, in CpuDevice() function in xla::CpuDevice
/external/tensorflow/tensorflow/core/framework/
Ddevice_base_test.cc28 TEST(DeviceBaseTest, CpuDevice) { in TEST() argument
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dimplementation_selector_test.cc39 constexpr char CpuDevice[] = "/device:CPU:0"; variable