Searched defs:n_rounds (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_inductor/runtime/ |
| H A D | halide_helpers.py | 45 def philox_impl(c0, c1, c2, c3, k0, k1, n_rounds): argument
65 def halide_philox(seed, c0, c1, c2, c3, n_rounds): argument
76 def randint4x(seed, offset, n_rounds): argument
82 def rand4x(seed, offset, n_rounds=PHILOX_N_ROUNDS_DEFAULT): argument
91 def randint(seed, offset, n_rounds=PHILOX_N_ROUNDS_DEFAULT): argument
96 def rand(seed, offset, n_rounds=PHILOX_N_ROUNDS_DEFAULT): argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/core/ |
| H A D | PhiloxRNGEngine.h | 122 inline float randn(uint32_t n_rounds) { in randn()
225 …T_DEVICE inline detail::UINT4 rand(detail::UINT4& counter, detail::UINT2& key, uint32_t n_rounds) { in rand()
|