Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dxrt_state_ops.cc61 .Device(DEVICE_XLA_GPU)
72 .Device(DEVICE_XLA_GPU)
81 .Device(DEVICE_XLA_GPU)
92 .Device(DEVICE_XLA_GPU)
105 .Device(DEVICE_XLA_GPU)
118 .Device(DEVICE_XLA_GPU)
131 .Device(DEVICE_XLA_GPU)
142 .Device(DEVICE_XLA_GPU)
155 .Device(DEVICE_XLA_GPU)
166 .Device(DEVICE_XLA_GPU)
[all …]
Dxrt_compile_ops.cc286 .Device(DEVICE_XLA_GPU)
296 .Device(DEVICE_XLA_GPU)
Dxrt_execute_op.cc590 .Device(DEVICE_XLA_GPU)
605 .Device(DEVICE_XLA_GPU)
/external/tensorflow/tensorflow/compiler/jit/
Dxla_gpu_device.cc88 absl::StrCat("/physical_device:", DEVICE_XLA_GPU, ":", i)); in ListPhysicalDevices()
116 XlaOpRegistry::RegisterCompilationDevice(DEVICE_XLA_GPU, registration); in CreateDevices()
119 RegisterXlaDeviceKernels(DEVICE_XLA_GPU, DEVICE_GPU_XLA_JIT); in CreateDevices()
152 options.device_name = DEVICE_XLA_GPU; in CreateDevices()
171 REGISTER_LOCAL_DEVICE_FACTORY(DEVICE_XLA_GPU, XlaGpuDeviceFactory);
180 REGISTER_XLA_LAUNCH_KERNEL(DEVICE_XLA_GPU, XlaLocalLaunchOp, kAllXlaGpuTypes);
181 REGISTER_XLA_COMPILE_KERNEL(DEVICE_XLA_GPU, XlaCompileOp, kAllXlaGpuTypes);
182 REGISTER_XLA_RUN_KERNEL(DEVICE_XLA_GPU, XlaRunOp, kAllXlaGpuTypes);
184 REGISTER_XLA_DEVICE_KERNELS(DEVICE_XLA_GPU, kAllXlaGpuTypes);
/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.h46 extern const char* const DEVICE_XLA_GPU;
Dxla_op_registry.cc40 const char* const DEVICE_XLA_GPU = "XLA_GPU"; variable