Searched refs:RandomAlgorithmToAlgorithm (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | stateless_random_ops_v2.cc | 53 inline Algorithm RandomAlgorithmToAlgorithm(xla::RandomAlgorithm const& alg) { in RandomAlgorithmToAlgorithm() function
65 Algorithm alg_ = RandomAlgorithmToAlgorithm(alg); in GetCounter()
188 auto real_counter_size = GetCounterSize(RandomAlgorithmToAlgorithm(alg)); in MaybeSliceCounter()
218 OP_REQUIRES_OK(ctx, CheckKeyCounterShape(RandomAlgorithmToAlgorithm(alg), in Compile()
271 OP_REQUIRES_OK(ctx, CheckKeyCounterShape(RandomAlgorithmToAlgorithm(alg), in Compile()
332 OP_REQUIRES_OK(ctx, CheckKeyCounterShape(RandomAlgorithmToAlgorithm(alg), in Compile()
378 OP_REQUIRES_OK(ctx, CheckKeyCounterShape(RandomAlgorithmToAlgorithm(alg), in Compile()
432 OP_REQUIRES_OK(ctx, CheckKeyCounterShape(RandomAlgorithmToAlgorithm(alg), in Compile()
|