diff options
Diffstat (limited to 'tensorflow/compiler/xla/client/local_client.cc')
-rw-r--r-- | tensorflow/compiler/xla/client/local_client.cc | 20 |
1 files changed, 20 insertions, 0 deletions
diff --git a/tensorflow/compiler/xla/client/local_client.cc b/tensorflow/compiler/xla/client/local_client.cc index b50425a09c..15c744ecd3 100644 --- a/tensorflow/compiler/xla/client/local_client.cc +++ b/tensorflow/compiler/xla/client/local_client.cc @@ -27,6 +27,16 @@ namespace se = ::perftools::gputools; namespace xla { +ExecutableBuildOptions& ExecutableBuildOptions::set_platform( + perftools::gputools::Platform* platform) { + platform_ = platform; + return *this; +} + +perftools::gputools::Platform* ExecutableBuildOptions::platform() const { + return platform_; +} + ExecutableBuildOptions& ExecutableBuildOptions::set_device_ordinal( int device_ordinal) { device_ordinal_ = device_ordinal; @@ -46,6 +56,16 @@ const Shape* ExecutableBuildOptions::result_layout() const { return result_layout_set_ ? &result_layout_ : nullptr; } +ExecutableBuildOptions& ExecutableBuildOptions::set_has_hybrid_result( + bool has_hybrid_result) { + has_hybrid_result_ = has_hybrid_result; + return *this; +} + +bool ExecutableBuildOptions::has_hybrid_result() const { + return has_hybrid_result_; +} + namespace { StatusOr<Backend::StreamPtr> BorrowStreamForDevice(int device_ordinal, Backend* backend) { |