diff options
author | 2018-03-20 02:26:31 -0700 | |
---|---|---|
committer | 2018-03-20 02:30:28 -0700 | |
commit | 28db3a7eae4986e3e662de16188cf7a03be33768 (patch) | |
tree | 0e76009bf7acf6fa33dc8cbe42b15db2f5f8114c /tensorflow/compiler/jit/xla_launch_util.cc | |
parent | e2e67c528316be8ea4f624af8757e80d7f00b5b6 (diff) |
Fix bug
PiperOrigin-RevId: 189712233
Diffstat (limited to 'tensorflow/compiler/jit/xla_launch_util.cc')
-rw-r--r-- | tensorflow/compiler/jit/xla_launch_util.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/compiler/jit/xla_launch_util.cc b/tensorflow/compiler/jit/xla_launch_util.cc index 076cbd2084..bb7316c60c 100644 --- a/tensorflow/compiler/jit/xla_launch_util.cc +++ b/tensorflow/compiler/jit/xla_launch_util.cc @@ -169,7 +169,7 @@ void XlaComputationLaunchContext::PopulateOutputs( int output_num = 0; for (int i = 0; i < ctx->num_outputs(); ++i) { AllocatorAttributes alloc_attrs = ctx->output_alloc_attr(i); - Allocator* allocator = ctx->device()->GetAllocator(alloc_attrs); + Allocator* allocator = ctx->device()->GetAllocator({}); if (tensor_info_manager_ && !alloc_attrs.on_host()) { allocator = tensor_info_manager_; } |