Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_op_registry.cc114 CHECK(result.second || result.first->second.compilation_device_name == in RegisterCompilationDevice()
115 registration.compilation_device_name); in RegisterCompilationDevice()
119 const string& compilation_device_name, in RegisterBackend() argument
123 auto result = registry.backends_.emplace(compilation_device_name, Backend()); in RegisterBackend()
125 << compilation_device_name; in RegisterBackend()
146 registration.compilation_device_name = DEVICE_CPU_XLA_JIT; in GetCompilationDevice()
156 registration.compilation_device_name = DEVICE_GPU_XLA_JIT; in GetCompilationDevice()
325 const string& compilation_device_name, in DeviceKernels() argument
332 auto it = registry.backends_.find(compilation_device_name); in DeviceKernels()
334 << "Unknown backend " << compilation_device_name; in DeviceKernels()
Dxla_op_registry.h86 string compilation_device_name; member
105 static void RegisterBackend(const string& compilation_device_name,
138 const string& compilation_device_name,
/external/tensorflow/tensorflow/compiler/jit/
Dxla_interpreter_device.cc47 registration.compilation_device_name = DEVICE_INTERPRETER_XLA_JIT; in CreateDevices()
62 options.compilation_device_name = DEVICE_INTERPRETER_XLA_JIT; in CreateDevices()
Dxla_cpu_device.cc44 registration.compilation_device_name = DEVICE_CPU_XLA_JIT; in CreateDevices()
64 options.compilation_device_name = DEVICE_CPU_XLA_JIT; in CreateDevices()
Dxla_gpu_device.cc66 registration.compilation_device_name = DEVICE_GPU_XLA_JIT; in CreateDevices()
99 options.compilation_device_name = DEVICE_GPU_XLA_JIT; in CreateDevices()
Dxla_device.cc196 DeviceType(options.compilation_device_name), in XlaDevice()
202 jit_device_name_(options.compilation_device_name), in XlaDevice()
207 VLOG(1) << "Created XLA device " << options.compilation_device_name << " " in XlaDevice()
Dxla_device.h111 string compilation_device_name; member
Dmark_for_compilation_pass.cc504 DeviceType jit_device_type(registration->compilation_device_name); in FindCompilationCandidates()
658 DeviceType jit_device_type(registration->compilation_device_name); in IsCompilable()
/external/tensorflow/tensorflow/compiler/jit/kernels/
Dxla_ops.cc250 client.ValueOrDie(), DeviceType(registration->compilation_device_name)); in BuildCompilationCache()