diff options
Diffstat (limited to 'tensorflow/compiler/tf2xla/xla_compiler.h')
-rw-r--r-- | tensorflow/compiler/tf2xla/xla_compiler.h | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/tensorflow/compiler/tf2xla/xla_compiler.h b/tensorflow/compiler/tf2xla/xla_compiler.h index ac7d4cfb12..4d40ca5825 100644 --- a/tensorflow/compiler/tf2xla/xla_compiler.h +++ b/tensorflow/compiler/tf2xla/xla_compiler.h @@ -236,6 +236,12 @@ class XlaCompiler { // to the computation. bool allow_cpu_custom_calls = false; + // If 'local_executable_has_hybrid_result', the top-level pointers of the + // result tuple of compiled programs are stored in host memory and the + // nested buffers in device memory, otherwise the whole result tuple is + // stored in device memory. + bool local_executable_has_hybrid_result = false; + // If not nullptr, populate_resource_manager is called with the // compilation device's resource manager when the compilation // device is created, and can be used to create metadata objects |