diff options
Diffstat (limited to 'tensorflow/compiler/xla/service/hlo_module_config.h')
-rw-r--r-- | tensorflow/compiler/xla/service/hlo_module_config.h | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/tensorflow/compiler/xla/service/hlo_module_config.h b/tensorflow/compiler/xla/service/hlo_module_config.h index a5ee895e48..4a7ead9c10 100644 --- a/tensorflow/compiler/xla/service/hlo_module_config.h +++ b/tensorflow/compiler/xla/service/hlo_module_config.h @@ -104,6 +104,16 @@ class HloModuleConfig { // Whether to enable HLO-level profiling. bool hlo_profiling_enabled_ = false; + // If this flag is true, the generated executable will return a ShapedBuffer + // holding the result of the computation. In a ShapedBuffer, tuples have their + // structure held in host memory and the element arrays (leaves of the tuple + // structure) stored in device memory. The ShapedBuffer is considered "hybrid" + // because its leaves are on device but its structure is stored on + // host. Otherwise, if this flag is false, the generated executable will + // return a DeviceMemoryBase where the result is held entirely in device + // memory. + bool has_hybrid_result_ = false; + // Module/graph-level seed handle. uint64 seed_ = 0; |