#include "tensorflow/compiler/mlir/tensorflow/ir/tf_ops_layout_helper.h"

#include "mlir/Support/LLVM.h"  // from @llvm-project

namespace mlir {
namespace TF {

SmallVector<int64_t, 4> ReversePermutation(ArrayRef<int64_t> permutation) {
  SmallVector<int64_t, 4> reverse(permutation.size());
  for (size_t i = 0; i < permutation.size(); ++i) {
    reverse[permutation[i]] = i;
  }
  return reverse;
}