experiments
/
STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Merge pull request #40304 from ROCmSoftwarePlatform:google-upstream-non-max-supp-2 

PiperOrigin-RevId: 317683401
Change-Id: I537a9509e5701660470aa2059ff44e1277fde7e9
This commit is contained in:
TensorFlower Gardener 2020-06-22 11:37:55 -07:00
parent 3a22b091cc 4644b47db1
commit 197c34b17a
1 changed files with 0 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
tensorflow/core/kernels/non_max_suppression_op.cu.cc
View File

@ -28,25 +28,6 @@ limitations under the License.
#include "tensorflow/core/util/gpu_launch_config.h"
#include "tensorflow/stream_executor/stream_executor.h"
#define TF_RETURN_IF_CUDA_ERROR(result) \
do { \
cudaError_t error(result); \
if (!SE_PREDICT_TRUE(error == cudaSuccess)) { \
return errors::Internal("Cuda call failed with ", \
cudaGetErrorString(error)); \
} \
} while (0)
#define TF_OP_REQUIRES_CUDA_SUCCESS(context, result) \
do { \
cudaError_t error(result); \
if (!SE_PREDICT_TRUE(error == cudaSuccess)) { \
context->SetStatus(errors::Internal("Cuda call failed with", \
cudaGetErrorString(error))); \
return; \
} \
} while (0)
struct __align__(16) Box {
float x1, y1, x2, y2;
};