Home
last modified time | relevance | path

Searched refs:gpuEventSynchronize (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/image/
H A Dnon_max_suppression_op.cu.cc311 TF_RETURN_IF_CUDA_ERROR(gpuEventSynchronize(copy_done)); in CountIf()
774 TF_RETURN_IF_CUDA_ERROR(gpuEventSynchronize(copy_done)); in NmsGpu()
H A Dgenerate_box_proposals_op.cu.cc510 TF_OP_REQUIRES_CUDA_SUCCESS(context, gpuEventSynchronize(copy_done)); in Compute()
/aosp_15_r20/external/tensorflow/tensorflow/core/util/
H A Dgpu_device_functions.h48 #define gpuEventSynchronize cudaEventSynchronize macro
64 #define gpuEventSynchronize hipEventSynchronize macro