Searched refs:dev_image_boxes_keep_list (Results 1 – 1 of 1) sorted by relevance
224 Tensor* dev_image_prenms_scores, Tensor* dev_image_boxes_keep_list, in AllocatePreNMSTempTensors() argument241 dev_image_boxes_keep_list)); in AllocatePreNMSTempTensors()242 TF_RETURN_IF_ERROR(ResetTensor<int32>(dev_image_boxes_keep_list, d)); in AllocatePreNMSTempTensors()430 Tensor dev_image_boxes_keep_list; in Compute() local438 &dev_image_boxes_keep_list, &dev_postnms_rois, in Compute()447 int* d_image_boxes_keep_list = dev_image_boxes_keep_list.flat<int>().data(); in Compute()474 ResetTensor<int32>(&dev_image_boxes_keep_list, d)); in Compute()