Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/image/
Dnon_max_suppression_op.cu.cc358 TF_RETURN_IF_CUDA_ERROR(gpuEventSynchronize(copy_done)); in NmsGpu()
404 TF_RETURN_IF_CUDA_ERROR(gpuEventSynchronize(copy_done)); in CountIf()
Dgenerate_box_proposals_op.cu.cc510 TF_OP_REQUIRES_CUDA_SUCCESS(context, gpuEventSynchronize(copy_done)); in Compute()
/external/tensorflow/tensorflow/core/util/
Dgpu_device_functions.h51 #define gpuEventSynchronize cudaEventSynchronize macro
69 #define gpuEventSynchronize hipEventSynchronize macro