diff options
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. |