diff options
Diffstat (limited to 'tensorflow/compiler/xla/tests/hlo_test_base.h')
-rw-r--r-- | tensorflow/compiler/xla/tests/hlo_test_base.h | 12 |
1 files changed, 6 insertions, 6 deletions
diff --git a/tensorflow/compiler/xla/tests/hlo_test_base.h b/tensorflow/compiler/xla/tests/hlo_test_base.h index 21d77c0cc4..0ae4bdc104 100644 --- a/tensorflow/compiler/xla/tests/hlo_test_base.h +++ b/tensorflow/compiler/xla/tests/hlo_test_base.h @@ -115,16 +115,16 @@ class HloTestBase : public ::testing::Test { } // Executes the given module and return the result as a Literal. - StatusOr<std::unique_ptr<Literal>> Execute( - std::unique_ptr<HloModule> module, absl::Span<Literal* const> arguments); + StatusOr<Literal> Execute(std::unique_ptr<HloModule> module, + absl::Span<Literal* const> arguments); // Same as above, except the module will be executed without running any HLO // passes on it. - std::unique_ptr<Literal> ExecuteNoHloPasses( - std::unique_ptr<HloModule> module, absl::Span<Literal* const> arguments); + Literal ExecuteNoHloPasses(std::unique_ptr<HloModule> module, + absl::Span<Literal* const> arguments); - std::unique_ptr<Literal> ExecuteAndTransfer( - std::unique_ptr<HloModule> module, absl::Span<Literal* const> arguments); + Literal ExecuteAndTransfer(std::unique_ptr<HloModule> module, + absl::Span<Literal* const> arguments); // Executes the given hlo module on two backends and compares results. // |