HomeSort by relevance Sort by last modified time
    Searched refs:InfeedThunk (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
infeed_thunk.h 32 class InfeedThunk : public Thunk {
34 // Constructs a InfeedThunk that copies data from the on-device
38 InfeedThunk(tensorflow::gtl::ArraySlice<BufferAllocation::Slice>
43 InfeedThunk(const InfeedThunk&) = delete;
44 InfeedThunk& operator=(const InfeedThunk&) = delete;
infeed_thunk.cc 24 InfeedThunk::InfeedThunk(
33 Status InfeedThunk::ExecuteOnStream(const BufferAllocations& buffer_allocations,
ir_emitter_unnested.cc     [all...]

Completed in 138 milliseconds