| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | thunk_schedule.cc | 30 const Thunk& thunk, const HloInstruction& operand, in AddDependenciesOnTransitiveOperands() argument
37 stream_assignment_->StreamNumberForHlo(*thunk_to_hlo_.at(&thunk))) { in AddDependenciesOnTransitiveOperands()
38 depends_on_[&thunk].push_back(FindOrDie(hlo_to_thunk, &operand)); in AddDependenciesOnTransitiveOperands()
44 AddDependenciesOnTransitiveOperands(thunk, *operand_of_operand, in AddDependenciesOnTransitiveOperands()
57 for (auto& thunk : *thunks_) { in ThunkSchedule()
58 thunk_total_order_.push_back(thunk.get()); in ThunkSchedule()
62 for (const auto& thunk : *thunks_) { in ThunkSchedule()
63 InsertOrDie(&hlo_to_thunk, thunk_to_hlo_.at(thunk.get()), thunk.get()); in ThunkSchedule()
66 for (const Thunk* thunk : thunk_total_order_) { in ThunkSchedule() local
67 const auto* dst = thunk_to_hlo_.at(thunk); in ThunkSchedule()
[all …]
|
| D | thunk_schedule.h | 65 const std::list<const Thunk*>& DependsOn(const Thunk* thunk) const;
67 bool Depended(const Thunk* thunk) const { in Depended() argument
68 return depended_by_.contains(thunk); in Depended()
78 int StreamNumberForThunk(const Thunk* thunk) const { in StreamNumberForThunk() argument
80 return stream_assignment_->StreamNumberForHlo(*thunk_to_hlo_.at(thunk)); in StreamNumberForThunk()
97 const Thunk& thunk, const HloInstruction& operand,
|
| D | sequential_thunk.cc | 33 for (auto& thunk : thunks_) { in Initialize() local
34 TF_RETURN_IF_ERROR(thunk->Initialize(executable, executor)); in Initialize()
42 for (const auto& thunk : thunks_) { in ExecuteOnStream() local
43 ScopedAnnotation annotation([&] { return thunk->profile_annotation(); }); in ExecuteOnStream()
44 TF_RETURN_IF_ERROR(thunk->ExecuteOnStream(params)); in ExecuteOnStream()
|
| D | thunk_emitter.h | 34 virtual void AddThunkToThunkSequence(std::unique_ptr<Thunk> thunk) = 0;
51 void AddThunkToThunkSequence(std::unique_ptr<Thunk> thunk) { in AddThunkToThunkSequence() argument
52 return context_->AddThunkToThunkSequence(std::move(thunk)); in AddThunkToThunkSequence()
|
| D | gpu_executable.cc | 168 for (Thunk* thunk : thunk_schedule_->TotalOrder()) { in ExecuteThunks()
172 ScopedAnnotation annotation([&] { return thunk->profile_annotation(); }); in ExecuteThunks()
174 int32 stream_no = thunk_schedule_->StreamNumberForThunk(thunk); in ExecuteThunks()
178 for (const Thunk* dependency : thunk_schedule_->DependsOn(thunk)) { in ExecuteThunks()
182 VLOG(2) << "Executing the thunk for " << thunk->profile_annotation() in ExecuteThunks()
199 TF_RETURN_IF_ERROR(thunk->ExecuteOnStream(thunk_params)); in ExecuteThunks()
200 if (thunk_schedule_->Depended(thunk)) { in ExecuteThunks()
204 thunk_to_finish_event[thunk] = std::move(finish_event); in ExecuteThunks()
549 for (Thunk* thunk : thunk_schedule_->TotalOrder()) { in ExecuteAsyncOnStreamImpl()
550 TF_RETURN_IF_ERROR(thunk->Initialize(*this, executor)); in ExecuteAsyncOnStreamImpl()
|
| /external/grpc-grpc/src/python/grpcio/grpc/ |
| D | _interceptor.py | 27 def _continuation(self, thunk, index): argument
28 return lambda context: self._intercept_at(thunk, index, context)
30 def _intercept_at(self, thunk, index, context): argument
33 thunk = self._continuation(thunk, index + 1)
34 return interceptor.intercept_service(thunk, context)
36 return thunk(context)
38 def execute(self, thunk, context): argument
39 return self._intercept_at(thunk, 0, context)
191 def __init__(self, thunk, method, interceptor): argument
192 self._thunk = thunk
[all …]
|
| /external/llvm/test/Transforms/Inline/ |
| D | inline-musttail-varargs.ll | 3 ; We can't inline this thunk yet, but one day we will be able to. And when we
8 define linkonce_odr void @thunk(i8* %this, ...) {
15 call void (i8*, ...) @thunk(i8* %p, i32 42)
19 ; CHECK: call void (i8*, ...) @thunk(i8* %p, i32 42)
21 ; FIXME: Inline the thunk. This should be significantly easier than inlining
|
| /external/escapevelocity/src/main/java/com/google/escapevelocity/ |
| D | Macro.java | 101 Node thunk = parameterThunks.get(var); in getVar() local
102 if (thunk == null) { in getVar()
110 return thunk.evaluate(originalEvaluationContext); in getVar()
123 final Node thunk = parameterThunks.get(var); in setVar() local
124 if (thunk == null) { in setVar()
131 parameterThunks.put(var, thunk); in setVar()
|
| /external/llvm-project/llvm/test/Demangle/ |
| D | ms-thunks.test | 6 ; CHECK: [thunk]: public: virtual int __cdecl C::f`adjustor{16}'(void)
9 ; CHECK: [thunk]: public: virtual void * __cdecl Derived::`vector deleting dtor'`vtordisp{-4, 0}'(u…
12 ; CHECK: [thunk]: public: virtual void __thiscall simple::A::f`vtordispex{8, 8, -4, 8}'(void)
15 ; CHECK: [thunk]: __cdecl Base::`vcall'{8, {flat}}
|
| D | ms-templates-memptrs.test | 6 ; CHECK: void __cdecl CallMethod<struct NegativeNVOffset::C, {[thunk]: __thiscall NegativeNVOffset:…
12 ; CHECK: void __cdecl CallMethod<struct M, {[thunk]: __thiscall M::`vcall'{0, {flat}}, 0}>(struct M…
18 ; CHECK: void __cdecl CallMethod<struct O, {[thunk]: __thiscall O::`vcall'{0, {flat}}, 4}>(struct O…
24 ; CHECK: void __cdecl CallMethod<struct S, &[thunk]: __thiscall S::`vcall'{0, {flat}}>(struct S &)
33 ; CHECK: void __cdecl CallMethod<struct U, {[thunk]: __thiscall U::`vcall'{0, {flat}}, 0, 0, 0}>(st…
42 ; CHECK: void __cdecl CallMethod<struct V, {[thunk]: __thiscall V::`vcall'{0, {flat}}, 0, 0}>(struc…
|
| /external/llvm-project/llvm/test/Transforms/ArgumentPromotion/ |
| D | pr27568.ll | 10 ; CHECK-NEXT: call void @thunk()
14 call void @thunk()
21 ; CHECK-NEXT: invoke void @thunk()
31 invoke void @thunk()
44 declare void @thunk()
|
| /external/llvm-project/lld/test/ELF/ |
| D | aarch64-cortex-a53-843419-large2.s | 11 adrp x0, thunk
15 thunk: label
16 ldr x0, [x0, :got_lo12:thunk]
|
| D | ppc64-long-branch-rel14.s | 45 beq high # Need a thunk
46 beq+ high # Need a thunk
48 beq .text_high+16 # Need a thunk
62 bge .text_low+8 # Need a thunk
63 bgt .text_low+8 # Need a thunk
|
| D | ppc32-long-thunk.s | 67 bl high@local # Need a thunk
68 bl high@local # Need a thunk
69 bl high+32768@plt # Need a thunk
71 bl .text_high+16 # Need a thunk
89 bl .text_low+8 # Need a thunk
|
| D | ppc64-long-branch.s | 56 bl high # Need a thunk
57 bl high # Need a thunk
58 bl high # Need a thunk
60 bl .text_high+16 # Need a thunk
83 bl .text_low+8 # Need a thunk
|
| D | ppc64-long-branch-pi.s | 62 bl .text_high # Need a thunk
64 bl .text_high+8 # Need a thunk
65 bl .text_high+0xc # Need a thunk
83 bl .text_low # Need a thunk
|
| /external/llvm-project/lld/test/ELF/linkerscript/ |
| D | symbol-assign-many-passes2.test | 2 …ild-attributes -filetype=obj -triple=armv7a-linux-gnueabihf %S/Inputs/arm-thunk-many-passes.s -o %…
6 ## arm-thunk-many-passes.s is worst case case of thunk generation that takes 9
9 ## 1. we don't error that "thunk creation not converged".
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | tensor_cord_test.cc | 81 auto thunk = CreateThunkFor(cleaner); in TEST() local
85 TensorCord tc(a, thunk, &cleaner); in TEST()
148 auto thunk = CreateThunkFor(cleaner); in TEST() local
152 TensorCord tc(a, thunk, &cleaner); in TEST()
169 auto thunk = CreateThunkFor(cleaner); in TEST() local
171 TensorCord tc(a, thunk, &cleaner); in TEST()
189 auto thunk = CreateThunkFor(cleaner); in TEST() local
191 TensorCord tc(a, thunk, &cleaner); in TEST()
|
| /external/llvm-project/llvm/test/Transforms/Attributor/ArgumentPromotion/ |
| D | pr27568.ll | 11 ; CHECK-NEXT: call void @thunk()
15 call void @thunk()
22 ; CHECK-NEXT: invoke void @thunk()
32 invoke void @thunk()
45 declare void @thunk()
|
| /external/llvm-project/llvm/test/tools/llvm-cxxfilt/ |
| D | coff-import.test | 10 CHECK: import thunk for std::future<void>
12 CHECK-SPLIT: import thunk for std::future<void> import thunk for std::future<void>
|
| /external/llvm/test/Transforms/DeadArgElim/ |
| D | dead_vaargs.ll | 4 call void (i32, ...) @thunk(i32 %A, i64 47, double 1.000000e+00)
11 ; CHECK: call void (i32, ...) @thunk(i32 %A, i64 47, double 1.000000e+00)
17 define internal void @thunk(i32 %X, ...) {
21 ; CHECK-LABEL: define internal void @thunk(i32 %X, ...)
|
| D | funclet.ll | 6 call void @thunk()
12 invoke void @thunk()
27 declare void @thunk()
|
| /external/llvm-project/llvm/test/Transforms/DeadArgElim/ |
| D | dead_vaargs.ll | 4 call void (i32, ...) @thunk(i32 %A, i64 47, double 1.000000e+00)
11 ; CHECK: call void (i32, ...) @thunk(i32 %A, i64 47, double 1.000000e+00)
17 define internal void @thunk(i32 %X, ...) {
21 ; CHECK-LABEL: define internal void @thunk(i32 %X, ...)
|
| D | funclet.ll | 6 call void @thunk()
12 invoke void @thunk()
27 declare void @thunk()
|
| /external/llvm/test/Transforms/ArgumentPromotion/ |
| D | pr27568.ll | 6 call void @thunk()
12 invoke void @thunk()
29 declare void @thunk()
|