Searched refs:dev_postnms_rois (Results 1 – 1 of 1) sorted by relevance
225 Tensor* dev_postnms_rois, Tensor* dev_postnms_rois_probs, in AllocatePreNMSTempTensors() argument248 dev_postnms_rois)); in AllocatePreNMSTempTensors()249 TF_RETURN_IF_ERROR(ResetTensor<float>(dev_postnms_rois, d)); in AllocatePreNMSTempTensors()431 Tensor dev_postnms_rois; in Compute() local438 &dev_image_boxes_keep_list, &dev_postnms_rois, in Compute()