| /external/tensorflow/tensorflow/compiler/tests/ |
| D | stateless_random_ops_test.py | 86 seed_t = array_ops.placeholder(dtypes.int32, shape=[2])
88 stateless_random_get_key_counter_alg(seed_t))
92 y = sess.run(x, {seed_t: [0x12345678, 0xabcdef1]})
95 stateless_random_get_key_counter(seed_t))
100 y = sess.run(x, {seed_t: [0x12345678, 0xabcdef1]})
106 seed_t = array_ops.placeholder(dtypes.int32, shape=[2])
118 pure = stateless_op(shape, seed=seed_t, dtype=dtype)
120 seed_t: seed
132 seed_t = array_ops.placeholder(dtypes.int32, shape=[2])
134 shape=[1000], seed=seed_t, maxval=maxval, dtype=dtype)
[all …]
|
| D | categorical_op_test.py | 148 seed_t = array_ops.placeholder(dtypes.int32, shape=[2])
152 seed_t,
154 y = sess.run(x, {seed_t: [0x12345678, 0xabcdef12]})
162 seed_t = array_ops.placeholder(dtypes.int32, shape=[2])
167 logits, num_samples, seed=seed_t)
168 values = [(seed, pure.eval(feed_dict={seed_t: seed})) for seed in seeds]
184 seed_t = array_ops.placeholder(dtypes.int32, shape=[2])
188 seed=seed_t,
190 y = sess.run(x, {seed_t: [0x12345678, 0xabcdef1]})
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | stateless_random_ops_v2.cc | 204 const Tensor& seed_t = ctx->input(0); in Compute() local
205 OP_REQUIRES(ctx, seed_t.dims() == 1 && seed_t.dim_size(0) == 2, in Compute()
207 seed_t.shape().DebugString())); in Compute()
221 OP_REQUIRES_OK(ctx, GenerateKey(seed_t, &key, &counter)); in Compute()
233 const Tensor& seed_t = ctx->input(0); in Compute() local
234 OP_REQUIRES(ctx, seed_t.dims() == 1 && seed_t.dim_size(0) == 2, in Compute()
236 seed_t.shape().DebugString())); in Compute()
248 OP_REQUIRES_OK(ctx, GenerateKey(seed_t, &key, &counter)); in Compute()
|
| D | stateless_random_gamma_op.cc | 203 const Tensor& seed_t = context->input(1); in Compute() local
206 OP_REQUIRES(context, seed_t.dims() == 1 && seed_t.dim_size(0) == 2, in Compute()
208 seed_t.shape().DebugString())); in Compute()
217 OP_REQUIRES_OK(context, GenerateKey(seed_t, &key, &counter)); in Compute()
|
| D | stateless_random_ops.cc | 77 const Tensor& seed_t = context->input(1); in Compute() local
80 OP_REQUIRES(context, seed_t.dims() == 1 && seed_t.dim_size(0) == 2, in Compute()
82 seed_t.shape().DebugString())); in Compute()
91 OP_REQUIRES_OK(context, GenerateKey(seed_t, &key, &counter)); in Compute()
|
| D | multinomial_op.cc | 288 const Tensor& seed_t = ctx->input(2); in Compute() local
289 OP_REQUIRES(ctx, seed_t.dims() == 1 && seed_t.dim_size(0) == 2, in Compute()
291 seed_t.shape().DebugString())); in Compute()
295 OP_REQUIRES_OK(ctx, GenerateKey(seed_t, &key, &counter)); in Compute()
|
| D | stateless_random_ops.h | 29 Status GenerateKey(Tensor seed_t, random::PhiloxRandom::Key* out_key,
|
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | sample_distorted_bounding_box_op.cc | 464 const Tensor& seed_t = context->input(3); in Compute() local
465 OP_REQUIRES(context, seed_t.dims() == 1 && seed_t.dim_size(0) == 2, in Compute()
467 seed_t.shape().DebugString())); in Compute()
474 OP_REQUIRES_OK(context, GenerateKey(seed_t, &key, &counter)); in Compute()
|
| /external/tensorflow/tensorflow/python/kernel_tests/random/ |
| D | stateless_random_ops_test.py | 252 seed_t = array_ops.placeholder(seed_type, shape=[2])
253 pure = stateless_op(seed=seed_t)
255 (seed, pure.eval(feed_dict={seed_t: seed})) for seed in seeds
|