Searched refs:row_iou (Results 1 – 1 of 1) sorted by relevance
381 auto row_iou = xla::DynamicSlice(iou_mask, {row_idx, zero}, {1, num_boxes}); in operator ()() local384 row_iou = xla::DynamicUpdateSlice( in operator ()()385 row_iou, xla::ConstantR2FromArray2D<bool>(builder, {{false}}), in operator ()()388 row_iou = xla::Reshape(row_iou, {num_boxes}); in operator ()()389 auto supp_mask = xla::Not(row_iou); in operator ()()