Home
last modified time | relevance | path

Searched refs:outfeed_slices (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Doutfeed_thunk.cc26 OutfeedThunk::OutfeedThunk(ShapeTree<BufferAllocation::Slice> outfeed_slices, in OutfeedThunk() argument
29 outfeed_slices_(std::move(outfeed_slices)) {} in OutfeedThunk()
Doutfeed_thunk.h35 OutfeedThunk(ShapeTree<BufferAllocation::Slice> outfeed_slices,