HomeSort by relevance Sort by last modified time
    Searched refs:seed_t (Results 1 - 5 of 5) sorted by null

  /external/tensorflow/tensorflow/compiler/tests/
stateless_random_ops_test.py 41 seed_t = array_ops.placeholder(dtypes.int32, shape=[2])
48 pure = stateless_op(shape, seed=seed_t, dtype=dtype)
50 seed_t: seed
59 seed_t = array_ops.placeholder(dtypes.int32, shape=[2])
61 shape=[1000], seed=seed_t, dtype=dtype)
62 y = sess.run(x, {seed_t: [0x12345678, 0xabcdef12]})
78 seed_t = array_ops.placeholder(dtypes.int32, shape=[2])
81 shape=[n], seed=seed_t, dtype=dtype)
82 y = sess.run(x, {seed_t: [565656, 121212]})
106 seed_t = array_ops.placeholder(dtypes.int32, shape=[2]
    [all...]
  /external/tensorflow/tensorflow/contrib/stateless/python/kernel_tests/
stateless_random_ops_test.py 73 seed_t = array_ops.placeholder(seed_type, shape=[2])
77 pure = stateless_op(shape, seed=seed_t)
78 values = [(seed, pure.eval(feed_dict={seed_t: seed}))
87 seed_t = array_ops.placeholder(dtypes.int64, shape=[2])
92 seed=seed_t)
93 values = [(seed, pure.eval(feed_dict={seed_t: seed}))
  /external/tensorflow/tensorflow/core/kernels/
stateless_random_ops.cc 40 const Tensor& seed_t = context->input(1); variable
43 OP_REQUIRES(context, seed_t.dims() == 1 && seed_t.dim_size(0) == 2,
45 seed_t.shape().DebugString()));
56 const auto seed = seed_t.flat<int32>();
61 const auto seed = seed_t.flat<int64>();
  /external/tensorflow/tensorflow/contrib/boosted_trees/kernels/
prediction_ops.cc 234 const Tensor* seed_t; local
235 OP_REQUIRES_OK(context, context->input(kSeedTensorName, &seed_t));
236 OP_REQUIRES(context, TensorShapeUtils::IsScalar(seed_t->shape()),
238 const uint64 seed = seed_t->scalar<int64>()();
training_ops.cc 347 const Tensor* seed_t; variable
348 OP_REQUIRES_OK(context, context->input("dropout_seed", &seed_t));
350 const uint64 dropout_seed = seed_t->scalar<int64>()();
    [all...]

Completed in 537 milliseconds