Searched refs:last_blocking_thunk_for_stream (Results 1 - 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgpu_executable.cc165 // last_blocking_thunk_for_stream[N] completes.
166 std::map<int32, const Thunk*> last_blocking_thunk_for_stream; local
179 if (last_blocking_thunk_for_stream.count(stream_no)) {
181 last_blocking_thunk_for_stream[stream_no])
183 last_blocking_thunk_for_stream.erase(stream_no);
190 last_blocking_thunk_for_stream.clear();
205 // Set last_blocking_thunk_for_stream on all streams other than this one
210 last_blocking_thunk_for_stream[i] = thunk;

Completed in 210 milliseconds