(opr)
| 22 | |
| 23 | |
| 24 | def _is_rng_op(opr): |
| 25 | return isinstance( |
| 26 | opr, |
| 27 | ( |
| 28 | mops.Dropout, |
| 29 | mops.BetaRNG, |
| 30 | mops.GammaRNG, |
| 31 | mops.GaussianRNG, |
| 32 | mops.PermutationRNG, |
| 33 | mops.PoissonRNG, |
| 34 | mops.ShuffleRNG, |
| 35 | mops.UniformRNG, |
| 36 | mops.ExponentialRNG, |
| 37 | mops.MultinomialRNG, |
| 38 | ), |
| 39 | ) |
| 40 | |
| 41 | |
| 42 | class AbstractVar: |