Function GetPermutationIfAvailable

tensorflow/compiler/mlir/xla/hlo_utils.cc:44–54

}

llvm::SmallVector<AffineMap, 2> GetPermutationIfAvailable(
    const Shape& shape, mlir::Builder builder) {
  if (!shape.has_layout() || shape.layout().minor_to_major().empty()) {
    return {};
  }
  llvm::SmallVector<unsigned, 2> permutation;
  for (auto dim : llvm::reverse(shape.layout().minor_to_major())) {
    permutation.push_back(dim);
  }
  return {AffineMap::getPermutationMap(permutation, builder.getContext())};
}

}  // namespace