def METHOD_NAME(input: Store, bins: Store) -> Store:
    result = context.create_store(ty.uint64, input.shape)
    task = context.create_auto_task(OpCode.CATEGORIZE)
    task.add_input(input)
    task.add_input(bins)
    task.add_output(result)
    # Broadcast the store that contains bin edges. Each task will get a copy
    # of the entire bin edges
    task.add_broadcast(bins)
    task.execute()
    return result