Home
last modified time | relevance | path

Searched refs:deferred_host_callbacks (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dthunk.h111 std::vector<std::function<void()>>* deferred_host_callbacks; // never null member
134 std::vector<std::function<void()>>* deferred_host_callbacks) { in SafeH2DMemcpy() argument
137 deferred_host_callbacks->push_back([buf_raw] { delete[] buf_raw; }); in SafeH2DMemcpy()
Dgpu_executable.cc167 std::vector<std::function<void()>> deferred_host_callbacks; in ExecuteThunks() local
192 &deferred_host_callbacks, in ExecuteThunks()
209 if (!deferred_host_callbacks.empty()) { in ExecuteThunks()
210 auto fn = [deferred_host_callbacks{std::move(deferred_host_callbacks)}]() { in ExecuteThunks()
211 for (auto& callback : deferred_host_callbacks) { in ExecuteThunks()