Searched refs:AsGpuStream (Results 1 – 8 of 8) sorted by relevance
51 GpuStream* AsGpuStream(Stream* stream) { in AsGpuStream() function58 return AsGpuStream(stream)->gpu_stream(); in AsGpuStreamValue()
91 GpuStream* AsGpuStream(Stream* stream);
30 return gpu::AsGpuStream(stream); in AsCUDAStream()
680 return AsGpuEvent(event)->Record(AsGpuStream(stream)); in RecordEvent()684 if (GpuDriver::WaitStreamOnEvent(context_, AsGpuStream(stream)->gpu_stream(), in WaitForEvent()700 return AsGpuStream(stream)->Init(); in AllocateStream()704 GpuStream* cuda_stream = AsGpuStream(stream); in DeallocateStream()720 CUevent other_completed_event = *AsGpuStream(other)->completed_event(); in CreateStreamDependency()735 return AsGpuTimer(timer)->Start(AsGpuStream(stream)); in StartTimer()739 return AsGpuTimer(timer)->Stop(AsGpuStream(stream)); in StopTimer()
1800 if (!timer->Init() || !timer->Start(AsGpuStream(stream))) { in DoRnnForwardImpl()1873 if (!timer->Stop(AsGpuStream(stream))) { in DoRnnForwardImpl()1931 if (!timer->Init() || !timer->Start(AsGpuStream(stream))) { in DoRnnBackwardImpl()2018 if (!timer->Stop(AsGpuStream(stream))) { in DoRnnBackwardImpl()3080 if (!timer->Init() || !timer->Start(AsGpuStream(stream))) { in DoConvolve()3163 if (!timer->Stop(AsGpuStream(stream))) { in DoConvolve()3231 if (!timer->Init() || !timer->Start(AsGpuStream(stream))) { in DoFusedConvolveImpl()3282 if (!timer->Stop(AsGpuStream(stream))) { in DoFusedConvolveImpl()
2002 if (!timer->Init() || !timer->Start(AsGpuStream(stream))) { in DoBlasInternalImpl()2014 if (!timer->Stop(AsGpuStream(stream))) { in DoBlasInternalImpl()2034 if (!timer->Init() || !timer->Start(AsGpuStream(stream))) { in DoBlasInternalImpl()2046 if (!timer->Stop(AsGpuStream(stream))) { in DoBlasInternalImpl()2139 if (!timer->Init() || !timer->Start(AsGpuStream(stream))) { in DoBlasInternalImpl()2179 if (!timer->Stop(AsGpuStream(stream))) { in DoBlasInternalImpl()3716 if (!timer->Init() || !timer->Start(AsGpuStream(stream))) { in DoBlasInternalImpl()3729 if (!timer->Stop(AsGpuStream(stream))) { in DoBlasInternalImpl()
588 return AsGpuEvent(event)->Record(AsGpuStream(stream)); in RecordEvent()592 if (GpuDriver::WaitStreamOnEvent(context_, AsGpuStream(stream)->gpu_stream(), in WaitForEvent()608 return AsGpuStream(stream)->Init(); in AllocateStream()612 GpuStream* rocm_stream = AsGpuStream(stream); in DeallocateStream()628 GpuEventHandle other_completed_event = *AsGpuStream(other)->completed_event(); in CreateStreamDependency()643 return AsGpuTimer(timer)->Start(AsGpuStream(stream)); in StartTimer()647 return AsGpuTimer(timer)->Stop(AsGpuStream(stream)); in StopTimer()
2974 if (!timer->Start(AsGpuStream(stream))) { in DoConvolve()3060 if (!timer->Stop(AsGpuStream(stream))) { in DoConvolve()4660 timer->Start(AsGpuStream(stream)); in DoFusedConvolutionBiasActivationImpl()4684 timer->Stop(AsGpuStream(stream)); in DoFusedConvolutionBiasActivationImpl()4758 timer->Start(AsGpuStream(stream)); in DoFusedBatchNormActivationInferenceImpl()4779 timer->Stop(AsGpuStream(stream)); in DoFusedBatchNormActivationInferenceImpl()4869 timer->Start(AsGpuStream(stream)); in DoFusedBatchNormActivationForwardImpl()4891 timer->Stop(AsGpuStream(stream)); in DoFusedBatchNormActivationForwardImpl()4985 timer->Start(AsGpuStream(stream)); in DoFusedBatchNormActivationBackwardImpl()5009 timer->Stop(AsGpuStream(stream)); in DoFusedBatchNormActivationBackwardImpl()