Searched refs:AsGpuStream (Results 1 – 8 of 8) sorted by relevance
50 GpuStream* AsGpuStream(Stream* stream) { in AsGpuStream() function57 return AsGpuStream(stream)->gpu_stream(); in AsGpuStreamValue()
88 GpuStream* AsGpuStream(Stream* stream);
589 return AsGpuEvent(event)->Record(AsGpuStream(stream)); in RecordEvent()593 if (GpuDriver::WaitStreamOnEvent(context_, AsGpuStream(stream)->gpu_stream(), in WaitForEvent()609 return AsGpuStream(stream)->Init(); in AllocateStream()613 GpuStream* rocm_stream = AsGpuStream(stream); in DeallocateStream()629 GpuEventHandle other_completed_event = *AsGpuStream(other)->completed_event(); in CreateStreamDependency()644 return AsGpuTimer(timer)->Start(AsGpuStream(stream)); in StartTimer()648 return AsGpuTimer(timer)->Stop(AsGpuStream(stream)); in StopTimer()
2844 if (!timer->Start(AsGpuStream(stream))) { in DoConvolve()2925 if (!timer->Stop(AsGpuStream(stream))) { in DoConvolve()4064 timer->Start(AsGpuStream(stream)); in DoFusedConvolutionBiasActivationImpl()4088 timer->Stop(AsGpuStream(stream)); in DoFusedConvolutionBiasActivationImpl()4162 timer->Start(AsGpuStream(stream)); in DoFusedBatchNormActivationInferenceImpl()4183 timer->Stop(AsGpuStream(stream)); in DoFusedBatchNormActivationInferenceImpl()4273 timer->Start(AsGpuStream(stream)); in DoFusedBatchNormActivationForwardImpl()4295 timer->Stop(AsGpuStream(stream)); in DoFusedBatchNormActivationForwardImpl()4389 timer->Start(AsGpuStream(stream)); in DoFusedBatchNormActivationBackwardImpl()4413 timer->Stop(AsGpuStream(stream)); in DoFusedBatchNormActivationBackwardImpl()
30 return gpu::AsGpuStream(stream); in AsCUDAStream()
708 return AsGpuEvent(event)->Record(AsGpuStream(stream)); in RecordEvent()712 if (GpuDriver::WaitStreamOnEvent(context_, AsGpuStream(stream)->gpu_stream(), in WaitForEvent()728 return AsGpuStream(stream)->Init(); in AllocateStream()732 GpuStream* cuda_stream = AsGpuStream(stream); in DeallocateStream()748 CUevent other_completed_event = *AsGpuStream(other)->completed_event(); in CreateStreamDependency()763 return AsGpuTimer(timer)->Start(AsGpuStream(stream)); in StartTimer()767 return AsGpuTimer(timer)->Stop(AsGpuStream(stream)); in StopTimer()
1561 if (!timer->Init() || !timer->Start(AsGpuStream(stream))) { in DoRnnForwardImpl()1646 if (!timer->Stop(AsGpuStream(stream))) { in DoRnnForwardImpl()1704 if (!timer->Init() || !timer->Start(AsGpuStream(stream))) { in DoRnnBackwardImpl()1803 if (!timer->Stop(AsGpuStream(stream))) { in DoRnnBackwardImpl()2720 if (!timer->Init() || !timer->Start(AsGpuStream(stream))) { in DoConvolve()2909 if (!timer->Stop(AsGpuStream(stream))) { in DoConvolve()2973 if (!timer->Init() || !timer->Start(AsGpuStream(stream))) { in DoFusedConvolveImpl()3026 if (!timer->Stop(AsGpuStream(stream))) { in DoFusedConvolveImpl()
1807 if (!timer->Init() || !timer->Start(AsGpuStream(stream))) { in DoBlasGemvWithProfilingImpl()1819 if (!timer->Stop(AsGpuStream(stream))) { in DoBlasGemvWithProfilingImpl()1839 if (!timer->Init() || !timer->Start(AsGpuStream(stream))) { in DoBlasGemmWithProfilingImpl()1851 if (!timer->Stop(AsGpuStream(stream))) { in DoBlasGemmWithProfilingImpl()1929 if (!timer->Init() || !timer->Start(AsGpuStream(stream))) { in DoBlasGemmWithAlgorithmImpl()1968 if (!timer->Stop(AsGpuStream(stream))) { in DoBlasGemmWithAlgorithmImpl()