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

  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
while_thunk.cc 26 WhileThunk::WhileThunk(
35 // are logically "part of" this WhileThunk, and shouldn't be profiled
42 Status WhileThunk::Initialize(const GpuExecutable& executable,
50 Status WhileThunk::ExecuteOnStream(const BufferAllocations& buffer_allocations,
while_thunk.h 31 // WhileThunk implements the while instruction on GPU by invoking a thunk
33 // thunk sequence for the while 'body' computation. WhileThunk assumes that
37 // WhileThunk synchronizes the stream to test the result of the 'condition'
39 class WhileThunk : public Thunk {
41 // Constructs a WhileThunk to compute while instruction 'hlo'.
42 WhileThunk(const BufferAllocation::Slice& condition_result_buffer_index,
46 WhileThunk(const WhileThunk&) = delete;
47 WhileThunk& operator=(const WhileThunk&) = delete
    [all...]

Completed in 159 milliseconds