diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2017-10-13 02:10:57 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-10-13 02:15:19 -0700 |
commit | 2645045db26914e179be2e161134be7e9cd9002b (patch) | |
tree | 63fb350de2fb34b5bdf388e782cb313360486717 /tensorflow/compiler/xla/service/compiler.h | |
parent | 5acdad4209cc8ee1a5c9421cbe38b0f2538843eb (diff) |
Extend the ExecuteParallel service interface to allow multiple devices per computation.
PiperOrigin-RevId: 172071664
Diffstat (limited to 'tensorflow/compiler/xla/service/compiler.h')
-rw-r--r-- | tensorflow/compiler/xla/service/compiler.h | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tensorflow/compiler/xla/service/compiler.h b/tensorflow/compiler/xla/service/compiler.h index d5bd9214be..4c2d9600d9 100644 --- a/tensorflow/compiler/xla/service/compiler.h +++ b/tensorflow/compiler/xla/service/compiler.h @@ -114,7 +114,8 @@ class Compiler { // sequence of executable objects. virtual StatusOr<std::vector<std::unique_ptr<Executable>>> Compile( std::vector<std::unique_ptr<HloModule>> modules, - std::vector<perftools::gputools::StreamExecutor*> stream_exec) = 0; + std::vector<std::vector<perftools::gputools::StreamExecutor*>> + stream_exec) = 0; // Compiles the HLO module for ahead-of-time execution. This is intended for // use in static compilation. |