aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/compiler/xla/client/local_client.cc
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/compiler/xla/client/local_client.cc')
-rw-r--r--tensorflow/compiler/xla/client/local_client.cc20
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) {