HomeSort by relevance Sort by last modified time
    Searched refs:DEVICE_XLA_GPU (Results 1 - 7 of 7) sorted by null

  /external/tensorflow/tensorflow/compiler/xrt/kernels/
xrt_state_ops.cc 30 .Device(DEVICE_XLA_GPU)
41 .Device(DEVICE_XLA_GPU)
52 .Device(DEVICE_XLA_GPU)
65 .Device(DEVICE_XLA_GPU)
78 .Device(DEVICE_XLA_GPU)
91 .Device(DEVICE_XLA_GPU)
102 .Device(DEVICE_XLA_GPU)
115 .Device(DEVICE_XLA_GPU)
126 .Device(DEVICE_XLA_GPU)
137 .Device(DEVICE_XLA_GPU)
    [all...]
xrt_compile_ops.cc 247 .Device(DEVICE_XLA_GPU)
257 .Device(DEVICE_XLA_GPU)
xrt_execute_op.cc 295 .Device(DEVICE_XLA_GPU)
  /external/tensorflow/tensorflow/compiler/jit/
xla_gpu_device.cc 70 XlaOpRegistry::RegisterCompilationDevice(DEVICE_XLA_GPU, registration);
73 RegisterXlaDeviceKernels(DEVICE_XLA_GPU, DEVICE_GPU_XLA_JIT);
97 options.device_name = DEVICE_XLA_GPU;
116 REGISTER_LOCAL_DEVICE_FACTORY(DEVICE_XLA_GPU, XlaGpuDeviceFactory);
124 REGISTER_XLA_LAUNCH_KERNEL(DEVICE_XLA_GPU, XlaLocalLaunchOp, kAllXlaGpuTypes);
125 REGISTER_XLA_COMPILE_KERNEL(DEVICE_XLA_GPU, XlaCompileOp, kAllXlaGpuTypes);
126 REGISTER_XLA_RUN_KERNEL(DEVICE_XLA_GPU, XlaRunOp, kAllXlaGpuTypes);
128 REGISTER_XLA_DEVICE_KERNELS(DEVICE_XLA_GPU, kAllXlaGpuTypes);
  /external/tensorflow/tensorflow/tools/optimization/
gpu_optimization_pass_runner_main.cc 47 tensorflow::DEVICE_XLA_GPU));
  /external/tensorflow/tensorflow/compiler/tf2xla/
xla_op_registry.h 46 extern const char* const DEVICE_XLA_GPU;
xla_op_registry.cc 40 const char* const DEVICE_XLA_GPU = "XLA_GPU";

Completed in 6894 milliseconds