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

  /external/tensorflow/tensorflow/compiler/jit/
xla_interpreter_device.cc 47 registration.compilation_device_name = DEVICE_INTERPRETER_XLA_JIT;
62 options.compilation_device_name = DEVICE_INTERPRETER_XLA_JIT;
xla_cpu_device.cc 44 registration.compilation_device_name = DEVICE_CPU_XLA_JIT;
64 options.compilation_device_name = DEVICE_CPU_XLA_JIT;
xla_gpu_device.cc 66 registration.compilation_device_name = DEVICE_GPU_XLA_JIT;
99 options.compilation_device_name = DEVICE_GPU_XLA_JIT;
xla_device.cc 196 DeviceType(options.compilation_device_name),
202 jit_device_name_(options.compilation_device_name),
207 VLOG(1) << "Created XLA device " << options.compilation_device_name << " "
xla_device.h 111 string compilation_device_name; member in struct:tensorflow::XlaDevice::Options
mark_for_compilation_pass.cc 504 DeviceType jit_device_type(registration->compilation_device_name);
658 DeviceType jit_device_type(registration->compilation_device_name);
    [all...]
  /external/tensorflow/tensorflow/compiler/tf2xla/
xla_op_registry.cc 114 CHECK(result.second || result.first->second.compilation_device_name ==
115 registration.compilation_device_name);
119 const string& compilation_device_name,
123 auto result = registry.backends_.emplace(compilation_device_name, Backend());
125 << compilation_device_name;
146 registration.compilation_device_name = DEVICE_CPU_XLA_JIT;
156 registration.compilation_device_name = DEVICE_GPU_XLA_JIT;
325 const string& compilation_device_name,
332 auto it = registry.backends_.find(compilation_device_name);
334 << "Unknown backend " << compilation_device_name;
    [all...]
xla_op_registry.h 86 string compilation_device_name; member in struct:tensorflow::XlaOpRegistry::DeviceRegistration
95 // Registers an XLA backend. `compilation_device_name` is the name of the
105 static void RegisterBackend(const string& compilation_device_name,
135 // 'compilation_device_name'. Does not include kernels registered as
138 const string& compilation_device_name,
  /external/tensorflow/tensorflow/compiler/jit/kernels/
xla_ops.cc 250 client.ValueOrDie(), DeviceType(registration->compilation_device_name));

Completed in 50 milliseconds