Searched refs:current_streams (Results 1 – 6 of 6) sorted by relevance
| /external/pytorch/c10/xpu/ |
| D | XPUStream.cpp | 34 thread_local std::unique_ptr<StreamId[]> current_streams = nullptr; variable
125 if (current_streams) { in initXPUStreamsOnce()
132 current_streams = std::make_unique<StreamId[]>(num_gpus); in initXPUStreamsOnce()
139 current_streams[i] = in initXPUStreamsOnce()
244 return XPUStreamForId(device, current_streams[device]); in getCurrentXPUStream()
251 current_streams[stream.device_index()] = stream.id(); in setCurrentXPUStream()
|
| /external/pytorch/c10/cuda/ |
| D | CUDAStream.cpp | 169 static thread_local std::unique_ptr<StreamId[]> current_streams = nullptr; variable
233 if (current_streams) { in initCUDAStreamsOnce()
239 current_streams = std::make_unique<StreamId[]>(num_gpus); in initCUDAStreamsOnce()
241 current_streams[i] = makeStreamId(StreamIdType::DEFAULT, 0); in initCUDAStreamsOnce()
370 return CUDAStreamForId(device_index, current_streams[device_index]); in getCurrentCUDAStream()
375 current_streams[stream.device_index()] = stream.id(); in setCurrentCUDAStream()
|
| /external/webrtc/pc/ |
| D | media_session.h | 247 StreamParamsVec* current_streams,
258 StreamParamsVec* current_streams,
267 StreamParamsVec* current_streams,
289 StreamParamsVec* current_streams,
303 StreamParamsVec* current_streams,
315 StreamParamsVec* current_streams,
|
| D | media_session.cc | 401 StreamParamsVec* current_streams, in AddStreamParams() argument
416 StreamParams* param = GetStreamByIds(*current_streams, sender.track_id); in AddStreamParams()
434 current_streams->push_back(stream_param); in AddStreamParams()
672 StreamParamsVec* current_streams, in CreateContentOffer() argument
722 StreamParamsVec* current_streams, in CreateMediaContentOffer() argument
728 current_streams, offer, field_trials)) { in CreateMediaContentOffer()
734 rtp_extensions, ssrc_generator, current_streams, in CreateMediaContentOffer()
1371 StreamParamsVec* current_streams, in SetCodecsInAnswer() argument
1382 current_streams, answer, field_trials)) { in SetCodecsInAnswer()
1404 StreamParamsVec* current_streams, in CreateMediaContentAnswer() argument
[all …]
|
| D | sdp_offer_answer.cc | 737 bool CanAddLocalMediaStream(webrtc::StreamCollectionInterface* current_streams, in CanAddLocalMediaStream() argument
739 if (!new_stream || !current_streams) { in CanAddLocalMediaStream()
742 if (current_streams->find(new_stream->id()) != nullptr) { in CanAddLocalMediaStream()
|
| /external/pytorch/test/cpp_extensions/ |
| D | mtia_extension.cpp | 13 static thread_local std::array<c10::Stream, kMTIADeviceCount> current_streams = variable
50 return current_streams[d.index()]; in getStream()
190 return current_streams[device]; in getCurrentStream()
196 current_streams[stream.device_index()] = stream; in setCurrentStream()
|