Home
last modified time | relevance | path

Searched refs:DEVICE_CPU_XLA_JIT (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/compiler/jit/
Dxla_cpu_device.cc44 registration.compilation_device_name = DEVICE_CPU_XLA_JIT; in CreateDevices()
53 RegisterXlaDeviceKernels(DEVICE_XLA_CPU, DEVICE_CPU_XLA_JIT); in CreateDevices()
64 options.compilation_device_name = DEVICE_CPU_XLA_JIT; in CreateDevices()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_cpu_backend.cc32 REGISTER_XLA_BACKEND(DEVICE_CPU_XLA_JIT, kCpuAllTypes, CpuOpFilter);
Dxla_op_registry_test.cc57 .Device(DEVICE_CPU_XLA_JIT)
Dtf2xla.cc261 absl::StrCat("/device:", DEVICE_CPU_XLA_JIT)); in ConvertGraphToXla()
292 compiler_options.device_type = DeviceType(DEVICE_CPU_XLA_JIT); in ConvertGraphToXla()
Dxla_op_registry.cc37 const char* const DEVICE_CPU_XLA_JIT = "XLA_CPU_JIT"; variable
146 registration.compilation_device_name = DEVICE_CPU_XLA_JIT; in GetCompilationDevice()
Dxla_op_registry.h42 extern const char* const DEVICE_CPU_XLA_JIT; // "CPU_XLA_JIT"
Dxla_compiler_test.cc65 options.device_type = DeviceType(DEVICE_CPU_XLA_JIT); in DefaultOptions()
167 REGISTER_XLA_OP(Name("DummyDuplicateOp").Device(DEVICE_CPU_XLA_JIT),
/external/tensorflow/tensorflow/tools/optimization/
Dgpu_optimization_pass_runner_main.cc50 tensorflow::DEVICE_CPU_XLA_JIT)); in main()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dindex_ops_cpu.cc137 .Device(DEVICE_CPU_XLA_JIT)