OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:step_ids
(Results
1 - 5
of
5
) sorted by null
/external/tensorflow/tensorflow/contrib/seq2seq/ops/
beam_search_ops.cc
26
.Input("
step_ids
: T")
33
ShapeHandle
step_ids
, parent_ids, max_sequence_lengths, end_token;
35
//
step_ids
, parent_ids, and output are all shaped:
38
TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 3, &
step_ids
));
42
TF_RETURN_IF_ERROR(c->Merge(
step_ids
, parent_ids, &
step_ids
));
43
DimensionHandle batch_size = c->Dim(
step_ids
, 1);
46
ShapeHandle step_ids_prefix = c->Matrix(c->Dim(
step_ids
, 0), batch_size);
47
TF_RETURN_IF_ERROR(c->MergePrefix(
step_ids
, step_ids_prefix, &
step_ids
,
[
all
...]
/external/tensorflow/tensorflow/contrib/seq2seq/python/kernel_tests/
beam_search_ops_test.py
40
step_ids
= _transpose_batch_time(
48
step_ids
=
step_ids
,
59
step_ids
= _transpose_batch_time(
66
step_ids
=
step_ids
,
83
step_ids
= _transpose_batch_time(
92
step_ids
=
step_ids
,
107
step_ids
= np.random.randint
[
all
...]
/external/tensorflow/tensorflow/contrib/seq2seq/kernels/
beam_search_ops.h
32
typename TTypes<T, 3>::ConstTensor
step_ids
,
beam_search_ops_gpu.cu.cc
30
const int32 beam_width, const T*
step_ids
,
47
beams[initial_beam_ix] = ldg(
step_ids
+ initial_beam_ix);
58
beams[level_beam_ix] = ldg(
step_ids
+ level_parent_ix);
83
typename TTypes<T, 3>::ConstTensor
step_ids
,
98
step_ids
.data(),
beam_search_ops.cc
50
const Tensor&
step_ids
= ctx->input(0);
variable
54
const TensorShape& step_ids_shape =
step_ids
.shape();
57
errors::InvalidArgument("
step_ids
must be a 3-tensor, saw shape: ",
70
"
step_ids
.shape must match parent_ids.shape. but shapes are: ",
76
errors::InvalidArgument("batch size dimensions
step_ids
.shape[1] and "
83
typename TTypes<T, 3>::ConstTensor step_ids_t(
step_ids
.tensor<T, 3>());
108
TTypes<int32, 3>::ConstTensor
step_ids
,
128
step_ids
(max_seq_len_b - 1, batch, beam);
138
beams(level, batch, beam) =
step_ids
(level, batch, parent);
174
typename TTypes<T, 3>::ConstTensor
step_ids
, \
[
all
...]
Completed in 75 milliseconds