Searched refs:dev_image_prenms_boxes (Results 1 – 1 of 1) sorted by relevance
223 OpKernelContext* context, Tensor* dev_image_prenms_boxes, in AllocatePreNMSTempTensors() argument231 dev_image_prenms_boxes)); in AllocatePreNMSTempTensors()232 TF_RETURN_IF_ERROR(ResetTensor<float>(dev_image_prenms_boxes, d)); in AllocatePreNMSTempTensors()428 Tensor dev_image_prenms_boxes; in Compute() local437 context, &dev_image_prenms_boxes, &dev_image_prenms_scores, in Compute()445 float* d_image_prenms_boxes = dev_image_prenms_boxes.flat<float>().data(); in Compute()