diff options
Diffstat (limited to 'tensorflow/compiler/xla/service/gpu/outfeed_thunk.cc')
-rw-r--r-- | tensorflow/compiler/xla/service/gpu/outfeed_thunk.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/compiler/xla/service/gpu/outfeed_thunk.cc b/tensorflow/compiler/xla/service/gpu/outfeed_thunk.cc index 4c0f1421e9..7986e63f43 100644 --- a/tensorflow/compiler/xla/service/gpu/outfeed_thunk.cc +++ b/tensorflow/compiler/xla/service/gpu/outfeed_thunk.cc @@ -36,7 +36,7 @@ Status OutfeedThunk::ExecuteOnStream( auto op_profiler = profiler->MakeScopedInstructionProfiler(hlo_instruction()); OutfeedManager* outfeed_manager = GetOrCreateOutfeedManager(); ShapeTree<std::unique_ptr<OutfeedBuffer>>* outfeed_buffers = - outfeed_manager->BlockingGetNextOutfeedDestination(); + outfeed_manager->BlockingGetNextDestination(); // Nothing to be done for empty tuples. if (ShapeUtil::IsEmptyTuple(hlo_instruction()->operand(0)->shape())) { |