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.cc35 .Device(DEVICE_XLA_CPU)
46 .Device(DEVICE_XLA_CPU)
58 .Device(DEVICE_XLA_CPU)
71 .Device(DEVICE_XLA_CPU)
84 .Device(DEVICE_XLA_CPU)
96 .Device(DEVICE_XLA_CPU)
108 .Device(DEVICE_XLA_CPU)
120 .Device(DEVICE_XLA_CPU)
131 .Device(DEVICE_XLA_CPU)
141 .Device(DEVICE_XLA_CPU)
[all …]
Dxrt_compile_ops.cc242 .Device(DEVICE_XLA_CPU)
253 .Device(DEVICE_XLA_CPU)
Dxrt_execute_op.cc287 .Device(DEVICE_XLA_CPU)
/external/tensorflow/tensorflow/compiler/jit/
Dxla_cpu_device.cc50 XlaOpRegistry::RegisterCompilationDevice(DEVICE_XLA_CPU, registration); in CreateDevices()
53 RegisterXlaDeviceKernels(DEVICE_XLA_CPU, DEVICE_CPU_XLA_JIT); in CreateDevices()
62 options.device_name = DEVICE_XLA_CPU; in CreateDevices()
82 REGISTER_LOCAL_DEVICE_FACTORY(DEVICE_XLA_CPU, XlaCpuDeviceFactory);
91 REGISTER_XLA_LAUNCH_KERNEL(DEVICE_XLA_CPU, XlaLocalLaunchOp, kAllXlaCpuTypes);
92 REGISTER_XLA_COMPILE_KERNEL(DEVICE_XLA_CPU, XlaCompileOp, kAllXlaCpuTypes);
93 REGISTER_XLA_RUN_KERNEL(DEVICE_XLA_CPU, XlaRunOp, kAllXlaCpuTypes);
95 REGISTER_XLA_DEVICE_KERNELS(DEVICE_XLA_CPU, kAllXlaCpuTypes);
/external/tensorflow/tensorflow/tools/optimization/
Dgpu_optimization_pass_runner_main.cc44 tensorflow::DEVICE_XLA_CPU)); in main()
/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