onnxruntime
717e764e - Move Class CudnnDropout to cudnn_common.h (#1492)

Commit
6 years ago
Move Class CudnnDropout to cudnn_common.h (#1492) 1. Move non_max_suppression_test.cc to object_detection folder 2. Move Class CudnnDropout to cudnn_common.h so that can share it with other ops. Move the cuda memory allocation part out of CudnnDropout to avoid memory leak.
Author
Parents
Loading