Searched refs:DEVICE_CPU_XLA_JIT (Results 1 – 11 of 11) sorted by relevance
60 registration.compilation_device_name = DEVICE_CPU_XLA_JIT; in CreateDevices()77 RegisterXlaDeviceKernels(DEVICE_XLA_CPU, DEVICE_CPU_XLA_JIT); in CreateDevices()88 options.compilation_device_name = DEVICE_CPU_XLA_JIT; in CreateDevices()
64 DeviceType device_type = DeviceType(DEVICE_CPU_XLA_JIT); in TEST()
67 REGISTER_XLA_OP(Name("CompilableOp").Device(DEVICE_CPU_XLA_JIT),106 DeviceType device_type_ = DeviceType(DEVICE_CPU_XLA_JIT);
63 absl::StrCat("/device:", DEVICE_CPU_XLA_JIT)); in ConvertGraphToXla()72 compiler_options.device_type = DeviceType(DEVICE_CPU_XLA_JIT); in ConvertGraphToXla()
32 REGISTER_XLA_BACKEND(DEVICE_CPU_XLA_JIT, kCpuAllTypes, CpuOpFilter);
57 .Device(DEVICE_CPU_XLA_JIT)
37 const char* const DEVICE_CPU_XLA_JIT = "XLA_CPU_JIT"; variable159 registration.compilation_device_name = DEVICE_CPU_XLA_JIT; in GetCompilationDevice()
42 extern const char* const DEVICE_CPU_XLA_JIT; // "CPU_XLA_JIT"
73 options.device_type = DeviceType(DEVICE_CPU_XLA_JIT); in DefaultOptions()175 REGISTER_XLA_OP(Name("DummyDuplicateOp").Device(DEVICE_CPU_XLA_JIT),
87 device_type_string == DEVICE_CPU_XLA_JIT) { in GetKeyCounter()102 device_type_string == DEVICE_CPU_XLA_JIT) { in GetAlg()
44 device_type_string == DEVICE_CPU_XLA_JIT) { in GetBitGeneratorForDevice()