| 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 = "mhlo.rng"(%mu, %sigma, %shape) {rng_distribution = #mhlo.rng_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 |
| } |
