decagon-pytorch/src/torch_stablesort/torch_stablesort_cpu.h

#pragma once

#include <torch/extension.h>

#include <vector>
#include <tuple>

std::vector<torch::Tensor> dispatch_cpu(torch::Tensor input,
  int dim,
  bool descending,
  torch::optional<std::tuple<torch::Tensor, torch::Tensor>> out);