OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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