Searched refs:RNG_MAX_COUNTER_SIZE (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/ops/ |
| D | stateless_random_ops_v2.cc | 113 c->set_output(1, c->MakeShape({RNG_MAX_COUNTER_SIZE})); in __anon293c03d20202()
133 c->set_output(1, c->MakeShape({RNG_MAX_COUNTER_SIZE})); in __anon293c03d20302()
|
| D | stateful_random_ops.cc | 103 c->set_output(0, c->MakeShape({RNG_MAX_COUNTER_SIZE + RNG_KEY_SIZE})); in __anonae3f3ebb0302()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | rng_alg.h | 35 static constexpr int RNG_MAX_COUNTER_SIZE = 2; variable
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | decompose_resource_ops.cc | 130 {tensorflow::RNG_MAX_COUNTER_SIZE + tensorflow::RNG_KEY_SIZE}, in matchAndRewrite()
194 int pad = tensorflow::RNG_MAX_COUNTER_SIZE - counter_size; in matchAndRewrite()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | stateless_random_ops_v2.cc | 214 ctx->allocate_output(1, TensorShape({RNG_MAX_COUNTER_SIZE}), in Compute()
243 ctx->allocate_output(1, TensorShape({RNG_MAX_COUNTER_SIZE}), in Compute()
|
| D | stateful_random_ops.cc | 288 ctx, ctx->allocate_output(0, {RNG_MAX_COUNTER_SIZE + RNG_KEY_SIZE}, in Compute()
291 if (RNG_MAX_COUNTER_SIZE > GetCounterSize(alg)) { in Compute()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | stateful_random_ops.cc | 450 if (RNG_MAX_COUNTER_SIZE > counter_size) { in Compile()
455 output = PadRight(output, RNG_MAX_COUNTER_SIZE - counter_size); in Compile()
|
| D | stateless_random_ops_v2.cc | 92 xla::ShapeUtil::MakeShape(xla::U64, {RNG_MAX_COUNTER_SIZE}); in GetKeyCounter()
|