Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/image/
Dnon_max_suppression_op.cu.cc166 __shared__ float shared_i_areas[kNmsBlockDim]; in NMSKernel() local
178 shared_i_areas[threadIdx.x] = (box.x2 - box.x1) * (box.y2 - box.y1); in NMSKernel()
202 if (OverThreshold<float>(&i_box, &j_box, shared_i_areas[threadIdx.x], in NMSKernel()