Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dslice_op_cpu_impl.h27 using CpuDevice = Eigen::ThreadPoolDevice; variable
H A Dquantize_and_dequantize_op.cc36 using CpuDevice = ::Eigen::ThreadPoolDevice; typedef
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/runtime/
H A Dconversion_function.h27 class CpuDevice; variable
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/kernel/conversion/
H A Dconversion.h27 class CpuDevice; variable
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/image/
H A Dmirror_pad_op_cpu_impl.h27 using CpuDevice = Eigen::ThreadPoolDevice; variable
H A Dmirror_pad_op.cc149 using CpuDevice = Eigen::ThreadPoolDevice; typedef
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dcpu_device.cc27 CpuDevice::CpuDevice(int id, in CpuDevice() function in xla::CpuDevice
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Ddevice_base_test.cc28 TEST(DeviceBaseTest, CpuDevice) { in TEST() argument
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dimplementation_selector_test.cc39 constexpr char CpuDevice[] = "/device:CPU:0"; variable
/aosp_15_r20/packages/modules/NeuralNetworks/runtime/
DManager.cpp860 class CpuDevice : public Device { class