Home
last modified time | relevance | path

Searched refs:RNG_ALG_THREEFRY (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Drng_alg.h29 RNG_ALG_THREEFRY = 2, enumerator
39 } else if (alg == RNG_ALG_THREEFRY) { in GetCounterSize()
/external/tensorflow/tensorflow/compiler/tests/
Dstateful_random_ops_test.py62 ALGS = [random.RNG_ALG_PHILOX, random.RNG_ALG_THREEFRY]
122 g = random.Generator.from_seed(seed=0, alg=random.RNG_ALG_THREEFRY)
172 gen = random.Generator(state=state, alg=random.RNG_ALG_THREEFRY)
394 gen = random.Generator.from_seed(seed=1234, alg=random.RNG_ALG_THREEFRY)
414 var.handle, random.RNG_ALG_THREEFRY, shape)
420 var.handle, random.RNG_ALG_THREEFRY, shape)
426 var.handle, random.RNG_ALG_THREEFRY, shape)
/external/tensorflow/tensorflow/python/ops/
Dstateful_random_ops.py73 RNG_ALG_THREEFRY = Algorithm.THREEFRY.value variable
142 elif alg == RNG_ALG_THREEFRY:
151 elif alg == RNG_ALG_THREEFRY:
187 return RNG_ALG_THREEFRY
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.py203 ("threefry", random.RNG_ALG_THREEFRY, random.Algorithm.THREEFRY)])
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Ddecompose_resource_ops.cc122 } else if (tensorflow::RNG_ALG_THREEFRY == alg_value) { in matchAndRewrite()
123 alg = tensorflow::RNG_ALG_THREEFRY; in matchAndRewrite()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dstateful_random_ops.cc140 if (alg == RNG_ALG_THREEFRY) { in StateAndKeyFromVariable()
163 if (alg == RNG_ALG_THREEFRY) { in StateAndKeyToVariable()
185 if (!(alg == RNG_ALG_THREEFRY || alg == RNG_ALG_PHILOX)) { in CompileImpl()
444 OP_REQUIRES(ctx, alg == RNG_ALG_THREEFRY || alg == RNG_ALG_PHILOX, in Compile()
Dstateless_random_ops_v2.cc45 } else if (alg == RNG_ALG_THREEFRY) { in AlgorithmToRandomAlgorithm()
57 return RNG_ALG_THREEFRY; in RandomAlgorithmToAlgorithm()
61 return RNG_ALG_THREEFRY; in RandomAlgorithmToAlgorithm()