Searched refs:iou_mask (Results 1 – 1 of 1) sorted by relevance
365 auto iou_mask = values[2]; in operator ()() local381 auto row_iou = xla::DynamicSlice(iou_mask, {row_idx, zero}, {1, num_boxes}); in operator ()()394 return std::vector<xla::XlaOp>{row_idx, num_outputs_so_far, iou_mask, in operator ()()