diff options
Diffstat (limited to 'tensorflow/compiler/jit/xla_compilation_cache.cc')
-rw-r--r-- | tensorflow/compiler/jit/xla_compilation_cache.cc | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/tensorflow/compiler/jit/xla_compilation_cache.cc b/tensorflow/compiler/jit/xla_compilation_cache.cc index bc2eccd277..23368b6c76 100644 --- a/tensorflow/compiler/jit/xla_compilation_cache.cc +++ b/tensorflow/compiler/jit/xla_compilation_cache.cc @@ -227,7 +227,10 @@ Status XlaCompilationCache::BuildExecutable( } xla::ExecutableBuildOptions build_options; build_options.set_device_ordinal(client_->default_device_ordinal()); + build_options.set_platform(client_->platform()); build_options.set_result_layout(result.xla_output_shape); + build_options.set_has_hybrid_result( + options.local_executable_has_hybrid_result); auto compile_result = client_->Compile(*result.computation, argument_layouts, build_options); |