OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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