OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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