Home
last modified time | relevance | path

Searched refs:seed_t (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/tests/
Dstateless_random_ops_test.py45 seed_t = array_ops.placeholder(dtypes.int32, shape=[2])
57 pure = stateless_op(shape, seed=seed_t, dtype=dtype)
59 seed_t: seed
71 seed_t = array_ops.placeholder(dtypes.int32, shape=[2])
73 shape=[1000], seed=seed_t, maxval=maxval, dtype=dtype)
74 y = sess.run(x, {seed_t: [0x12345678, 0xabcdef12]})
90 seed_t = array_ops.placeholder(dtypes.int32, shape=[2])
96 shape=[n], seed=seed_t, maxval=maxval, dtype=dtype)
97 y = sess.run(x, {seed_t: [565656, 121212]})
110 seed_t = array_ops.placeholder(dtypes.int32, shape=[2])
[all …]
Dcategorical_op_test.py147 seed_t = array_ops.placeholder(dtypes.int32, shape=[2])
151 seed_t,
153 y = sess.run(x, {seed_t: [0x12345678, 0xabcdef12]})
161 seed_t = array_ops.placeholder(dtypes.int32, shape=[2])
166 logits, num_samples, seed=seed_t)
167 values = [(seed, pure.eval(feed_dict={seed_t: seed})) for seed in seeds]
183 seed_t = array_ops.placeholder(dtypes.int32, shape=[2])
187 seed=seed_t,
189 y = sess.run(x, {seed_t: [0x12345678, 0xabcdef12]})
/external/tensorflow/tensorflow/core/kernels/
Dstateless_random_ops.cc75 const Tensor& seed_t = context->input(1); in Compute() local
78 OP_REQUIRES(context, seed_t.dims() == 1 && seed_t.dim_size(0) == 2, in Compute()
80 seed_t.shape().DebugString())); in Compute()
89 OP_REQUIRES_OK(context, GenerateKey(seed_t, &key, &counter)); in Compute()
Dmultinomial_op.cc288 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()
Dstateless_random_ops.h29 Status GenerateKey(Tensor seed_t, random::PhiloxRandom::Key* out_key,
/external/tensorflow/tensorflow/python/kernel_tests/random/
Dstateless_random_ops_test.py73 seed_t = array_ops.placeholder(seed_type, shape=[2])
76 pure = stateless_op(seed=seed_t)
78 (seed, pure.eval(feed_dict={seed_t: seed})) for seed in seeds
/external/tensorflow/tensorflow/contrib/boosted_trees/kernels/
Dprediction_ops.cc242 const Tensor* seed_t; in DoCompute() local
243 OP_REQUIRES_OK(context, context->input(kSeedTensorName, &seed_t)); in DoCompute()
244 OP_REQUIRES(context, TensorShapeUtils::IsScalar(seed_t->shape()), in DoCompute()
246 const uint64 seed = seed_t->scalar<int64>()(); in DoCompute()
Dtraining_ops.cc359 const Tensor* seed_t; in Compute() local
360 OP_REQUIRES_OK(context, context->input("dropout_seed", &seed_t)); in Compute()
362 const uint64 dropout_seed = seed_t->scalar<int64>()(); in Compute()