def METHOD_NAME(boxes_shape, scores_shape):
    out_shape = output_tensor((2,), "int64")
    count_shape = output_tensor((1,), "int64")
    out_shape[0] = boxes_shape[0] * scores_shape[1] * boxes_shape[1]
    out_shape[1] = int64(3)
    count_shape[0] = int64(1)
    return out_shape, count_shape