Searched refs:ThunkSequence (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | thunk_schedule.h | 54 std::unique_ptr<ThunkSequence> thunks,
59 explicit ThunkSchedule(std::unique_ptr<ThunkSequence> thunks);
100 std::unique_ptr<ThunkSequence> thunks_;
|
| D | while_thunk.h | 44 std::unique_ptr<ThunkSequence> condition_thunk_sequence,
45 std::unique_ptr<ThunkSequence> body_thunk_sequence,
|
| D | while_thunk.cc | 29 std::unique_ptr<ThunkSequence> condition_thunk_sequence, in WhileThunk()
30 std::unique_ptr<ThunkSequence> body_thunk_sequence, in WhileThunk()
|
| D | thunk_schedule.cc | 51 std::unique_ptr<ThunkSequence> thunks, in ThunkSchedule()
84 ThunkSchedule::ThunkSchedule(std::unique_ptr<ThunkSequence> thunks) in ThunkSchedule()
|
| D | for_thunk.h | 35 std::unique_ptr<ThunkSequence> body_thunk_sequence,
|
| D | for_thunk.cc | 27 std::unique_ptr<ThunkSequence> body_thunk_sequence, in ForThunk()
|
| D | conditional_thunk.h | 42 std::vector<ThunkSequence> branch_thunk_sequences,
|
| D | gpu_compiler.cc | 612 ThunkSequence thunk_sequence; in CompileModuleToLlvmIrImpl()
617 std::unique_ptr<ThunkSequence> thunks = in CompileModuleToLlvmIrImpl()
654 std::make_unique<ThunkSequence>(std::move(thunk_sequence)), in CompileModuleToLlvmIrImpl()
1121 ThunkSequence thunk_sequence; in CompileLmhloToExecutable()
1127 std::unique_ptr<ThunkSequence> thunks = ir_emitter->ConsumeThunkSequence(); in CompileLmhloToExecutable()
1135 std::make_unique<ThunkSequence>(std::move(thunk_sequence))); in CompileLmhloToExecutable()
|
| D | conditional_thunk.cc | 29 std::vector<ThunkSequence> branch_thunk_sequences, in GetConditionalThunkConfig()
|
| D | thunk.h | 147 using ThunkSequence = std::vector<std::unique_ptr<Thunk>>; variable
|
| D | ir_emitter_unnested.h | 155 std::unique_ptr<ThunkSequence> ConsumeThunkSequence() { in ConsumeThunkSequence()
156 return std::make_unique<ThunkSequence>(std::move(thunk_sequence_)); in ConsumeThunkSequence()
741 ThunkSequence thunk_sequence_;
|
| D | ir_emitter_unnested.cc | 3901 std::vector<ThunkSequence> branch_thunks; in BuildConditionalThunk()
|