Home
last modified time | relevance | path

Searched refs:RNG_ALG_PHILOX (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/compiler/tests/
Dstateful_random_ops_test.py62 ALGS = [random.RNG_ALG_PHILOX, random.RNG_ALG_THREEFRY]
143 g = random.Generator.from_seed(seed=0, alg=random.RNG_ALG_PHILOX)
192 gen = random.Generator(state=state, alg=random.RNG_ALG_PHILOX)
205 gen = random.Generator(state=state, alg=random.RNG_ALG_PHILOX)
222 seed=seed, alg=random.RNG_ALG_PHILOX)
225 seed=seed, alg=random.RNG_ALG_PHILOX)
244 seed=seed, alg=random.RNG_ALG_PHILOX).uniform_full_int(
249 seed=seed, alg=random.RNG_ALG_PHILOX).uniform_full_int(
265 seed=seed, alg=random.RNG_ALG_PHILOX)
268 seed=seed, alg=random.RNG_ALG_PHILOX)
[all …]
/external/tensorflow/tensorflow/core/framework/
Drng_alg.h25 RNG_ALG_PHILOX = 1, enumerator
37 if (alg == RNG_ALG_PHILOX) { in GetCounterSize()
/external/tensorflow/tensorflow/python/ops/
Dstateful_random_ops.py72 RNG_ALG_PHILOX = Algorithm.PHILOX.value variable
74 DEFAULT_ALGORITHM = RNG_ALG_PHILOX
140 if alg == RNG_ALG_PHILOX:
149 if alg == RNG_ALG_PHILOX:
185 return RNG_ALG_PHILOX
570 if alg == RNG_ALG_PHILOX or alg == RNG_ALG_THREEFRY:
925 if alg == RNG_ALG_PHILOX or alg == RNG_ALG_THREEFRY:
977 if alg == RNG_ALG_PHILOX or alg == RNG_ALG_THREEFRY:
Dstateful_random_ops_test.py75 random.RNG_ALG_PHILOX)
176 alg = random.RNG_ALG_PHILOX
202 ("philox", random.RNG_ALG_PHILOX, random.Algorithm.PHILOX),
216 alg = random.RNG_ALG_PHILOX
359 gen = random.Generator(state=[0, 0, key], alg=random.RNG_ALG_PHILOX)
372 gen = random.Generator(state=[counter, 0, key], alg=random.RNG_ALG_PHILOX)
385 alg=random.RNG_ALG_PHILOX)
551 var.handle, random.RNG_ALG_PHILOX, shape)
557 var.handle, random.RNG_ALG_PHILOX, shape)
563 var.handle, random.RNG_ALG_PHILOX, shape)
/external/tensorflow/tensorflow/core/kernels/
Dstateless_random_ops_v2.cc113 if (alg == RNG_ALG_PHILOX) { in Fill()
158 if (alg == RNG_ALG_PHILOX) { in Fill()
185 if (alg == RNG_ALG_PHILOX) { in Fill()
224 alg_output->flat<int>()(0) = RNG_ALG_PHILOX; in Compute()
261 alg_output->flat<int>()(0) = RNG_ALG_PHILOX; in Compute()
Dstateful_random_ops.cc108 if (alg == RNG_ALG_PHILOX) { in UpdateVariableAndFill()
150 RNG_ALG_PHILOX /*dummy*/); in Compute()
299 if (alg == RNG_ALG_PHILOX) { in Compute()
Drandom_binomial_op.cc410 OP_REQUIRES(ctx, alg == RNG_ALG_PHILOX, in Compute()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dstateful_random_ops.cc42 if (alg == RNG_ALG_PHILOX) { in BitGen()
118 if (alg == RNG_ALG_PHILOX) { in GetMinStateSize()
185 if (!(alg == RNG_ALG_THREEFRY || alg == RNG_ALG_PHILOX)) { in CompileImpl()
415 if (alg == RNG_ALG_PHILOX) { in IncreaseCounter()
444 OP_REQUIRES(ctx, alg == RNG_ALG_THREEFRY || alg == RNG_ALG_PHILOX, in Compile()
Dstateless_random_ops_v2.cc43 if (alg == RNG_ALG_PHILOX) { in AlgorithmToRandomAlgorithm()
55 return RNG_ALG_PHILOX; in RandomAlgorithmToAlgorithm()
103 return RNG_ALG_PHILOX; in GetAlg()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Ddecompose_resource_ops.cc120 if (tensorflow::RNG_ALG_PHILOX == alg_value) { in matchAndRewrite()
121 alg = tensorflow::RNG_ALG_PHILOX; in matchAndRewrite()