module {
  func.func @main(%arg0: tensor<1x3x2x2xf32>) -> (tensor<1x3x2x2xf32>, tensor<1x3x2x2xf32>) {
    %cst = stablehlo.constant dense<0.317560464> : tensor<1x3x2x2xf32>
    %cst_0 = stablehlo.constant dense<1.23323452> : tensor<1x3x2x2xf32>
    %0 = stablehlo.multiply %cst_0, %arg0 : tensor<1x3x2x2xf32>
    %1 = stablehlo.add %cst, %0 : tensor<1x3x2x2xf32>
    %2 = stablehlo.transpose %1, dims = [3, 2, 1, 0] : (tensor<1x3x2x2xf32>) -> tensor<2x2x3x1xf32>
    %3 = stablehlo.cosine %2 : tensor<2x2x3x1xf32>
    %4 = stablehlo.sine %2 : tensor<2x2x3x1xf32>
    %5 = stablehlo.transpose %4, dims = [3, 2, 1, 0] : (tensor<2x2x3x1xf32>) -> tensor<1x3x2x2xf32>
    %6 = stablehlo.transpose %3, dims = [3, 2, 1, 0] : (tensor<2x2x3x1xf32>) -> tensor<1x3x2x2xf32>
    return %5, %6 : tensor<1x3x2x2xf32>, tensor<1x3x2x2xf32>
  }
}