Searched refs:dev_postnms_rois_probs (Results 1 – 1 of 1) sorted by relevance
225 Tensor* dev_postnms_rois, Tensor* dev_postnms_rois_probs, in AllocatePreNMSTempTensors() argument253 dev_postnms_rois_probs)); in AllocatePreNMSTempTensors()254 TF_RETURN_IF_ERROR(ResetTensor<float>(dev_postnms_rois_probs, d)); in AllocatePreNMSTempTensors()432 Tensor dev_postnms_rois_probs; in Compute() local439 &dev_postnms_rois_probs, &dev_prenms_nboxes, num_images, in Compute()