HomeSort by relevance Sort by last modified time
    Searched defs:OutfeedThunk (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
outfeed_thunk.h 31 class OutfeedThunk : public Thunk {
33 // Constructs a OutfeedThunk that copies data to the host-side
35 OutfeedThunk(ShapeTree<BufferAllocation::Slice> outfeed_slices,
38 OutfeedThunk(const OutfeedThunk&) = delete;
39 OutfeedThunk& operator=(const OutfeedThunk&) = delete;
outfeed_thunk.cc 26 OutfeedThunk::OutfeedThunk(ShapeTree<BufferAllocation::Slice> outfeed_slices,
31 Status OutfeedThunk::ExecuteOnStream(

Completed in 3266 milliseconds