#pragma once

#include <torch/csrc/jit/ir/ir.h>

namespace torch::jit {

// Eliminates common inputs among `aten::cat` ops.
TORCH_API bool EliminateConcatCommonInputs(const std::shared_ptr<Graph>& graph);

// Expands `aten::cat` ops into `aten::copy` ops and eliminates redudancies
// in the buffers used for concatenation if possible.
TORCH_API void ExpandConcatAndEliminateRedundancy(
    const std::shared_ptr<Graph>& graph);

TORCH_API bool CombineConcats(const std::shared_ptr<Graph>& graph);

} // namespace torch::jit