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

Add some changes to trigger copybara import again.

Browse Source
This commit is contained in:
Guangda Lai 2019-11-07 07:44:46 -08:00 committed by GitHub
parent fbd53c577a
commit b0f61d8fbb
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tensorflow/core/kernels/generate_box_proposals_op.cu.cc
View File

@ -13,8 +13,6 @@ See the License for the specific language governing permissions and
limitations under the License. limitations under the License.
==============================================================================*/ ==============================================================================*/
// An example Op.
#if GOOGLE_CUDA #if GOOGLE_CUDA
#define EIGEN_USE_GPU #define EIGEN_USE_GPU
@ -82,8 +80,7 @@ __global__ void GeneratePreNMSUprightBoxesKernel(
// box_conv_index : # of the same box, but indexed in the // box_conv_index : # of the same box, but indexed in the
// scores from the conv layer, of shape (height,width,num_anchors) the // scores from the conv layer, of shape (height,width,num_anchors) the
// num_images dimension was already removed box_conv_index = // num_images dimension was already removed box_conv_index =
// a*image_stride + h*width // a*image_stride + h*width + w
// + w
const int box_conv_index = const int box_conv_index =
d_sorted_scores_keys[image_index * image_stride + ibox]; d_sorted_scores_keys[image_index * image_stride + ibox];