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