From: A. Unique TensorFlower <gardener@tensorflow.org>
Date: Tue, 20 Mar 2018 09:26:31 +0000 (-0700)
Subject: Fix bug
X-Git-Tag: tflite-v0.1.7~145^2^2~20
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=28db3a7eae4986e3e662de16188cf7a03be33768;p=platform%2Fupstream%2Ftensorflow.git

Fix bug

PiperOrigin-RevId: 189712233
---

diff --git a/tensorflow/compiler/jit/xla_launch_util.cc b/tensorflow/compiler/jit/xla_launch_util.cc
index 076cbd2..bb7316c 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_;
     }