Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/image/
Dnon_max_suppression_op.cu.cc357 TF_RETURN_IF_CUDA_ERROR(gpuEventRecord(copy_done, device.stream())); in NmsGpu()
403 TF_RETURN_IF_CUDA_ERROR(gpuEventRecord(copy_done, device.stream())); in CountIf()
Dgenerate_box_proposals_op.cu.cc509 gpuEventRecord(copy_done, d.stream())); in Compute()
/external/tensorflow/tensorflow/core/util/
Dgpu_device_functions.h50 #define gpuEventRecord cudaEventRecord macro
67 #define gpuEventRecord hipEventRecord macro