tests/e2e/stablehlo_ops/rng_normal.mlir

func.func @rng_normal_2d() {
  %mu = util.unfoldable_constant dense<0.0> : tensor<f32>
  %sigma = util.unfoldable_constant dense<1.0> : tensor<f32>
  %shape = util.unfoldable_constant dense<[3, 5]>  : tensor<2xi64>
  %res = stablehlo.rng %mu, %sigma, %shape, distribution = NORMAL : (tensor<f32>, tensor<f32>, tensor<2xi64>) -> tensor<3x5xf32>
  check.expect_almost_eq_const(%res,
    dense<[[0.570861, 0.317593, -0.726538, 1.45925, -1.59632],
           [-0.639956, 0.703875, -0.8801, -0.848389, -0.453391],
           [0.645563, 0.543174, 0.2255, 0.0809385, -1.17198]]> : tensor<3x5xf32>) : tensor<3x5xf32>
  return
}