Home
last modified time | relevance | path

Searched refs:DEVICE_XLA_CPU (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dxrt_state_ops.cc66 .Device(DEVICE_XLA_CPU)
76 .Device(DEVICE_XLA_CPU)
86 .Device(DEVICE_XLA_CPU)
98 .Device(DEVICE_XLA_CPU)
111 .Device(DEVICE_XLA_CPU)
124 .Device(DEVICE_XLA_CPU)
136 .Device(DEVICE_XLA_CPU)
148 .Device(DEVICE_XLA_CPU)
160 .Device(DEVICE_XLA_CPU)
171 .Device(DEVICE_XLA_CPU)
[all …]
Dxrt_compile_ops.cc281 .Device(DEVICE_XLA_CPU)
292 .Device(DEVICE_XLA_CPU)
Dxrt_execute_op.cc582 .Device(DEVICE_XLA_CPU)
598 .Device(DEVICE_XLA_CPU)
/external/tensorflow/tensorflow/compiler/jit/
Dxla_cpu_device.cc45 devices->push_back(absl::StrCat("/physical_device:", DEVICE_XLA_CPU, ":0")); in ListPhysicalDevices()
74 XlaOpRegistry::RegisterCompilationDevice(DEVICE_XLA_CPU, registration); in CreateDevices()
77 RegisterXlaDeviceKernels(DEVICE_XLA_CPU, DEVICE_CPU_XLA_JIT); in CreateDevices()
86 options.device_name = DEVICE_XLA_CPU; in CreateDevices()
106 REGISTER_LOCAL_DEVICE_FACTORY(DEVICE_XLA_CPU, XlaCpuDeviceFactory);
115 REGISTER_XLA_LAUNCH_KERNEL(DEVICE_XLA_CPU, XlaLocalLaunchOp, kAllXlaCpuTypes);
116 REGISTER_XLA_COMPILE_KERNEL(DEVICE_XLA_CPU, XlaCompileOp, kAllXlaCpuTypes);
117 REGISTER_XLA_RUN_KERNEL(DEVICE_XLA_CPU, XlaRunOp, kAllXlaCpuTypes);
119 REGISTER_XLA_DEVICE_KERNELS(DEVICE_XLA_CPU, kAllXlaCpuTypes);
/external/tensorflow/tensorflow/compiler/tests/
Dunary_ops_composition_test.cc58 tensorflow::IsGoogleCudaEnabled() ? DEVICE_XLA_GPU : DEVICE_XLA_CPU; in RunComposedOp()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_op_registry.h45 extern const char* const DEVICE_XLA_CPU;
Dxla_op_registry.cc39 const char* const DEVICE_XLA_CPU = "XLA_CPU"; variable