| /external/tensorflow/tensorflow/contrib/seq2seq/python/kernel_tests/ |
| beam_search_ops_test.py | 39 end_token = 10
51 end_token=end_token)
58 end_token = 10
69 end_token=end_token)
82 end_token = 10
95 end_token=end_token)
104 end_token =
[all...] |
| beam_search_decoder_test.py | 66 end_token=11)
114 self.end_token = 0
148 end_token=self.end_token,
203 end_token=self.end_token,
238 self.end_token = 0
301 end_token=self.end_token,
331 end_token = vocab_size -
[all...] |
| basic_decoder_test.py | 137 end_token = 1
144 end_token)
191 expected_step_finished = (expected_sample_ids == end_token)
210 end_token = 1
220 end_token, seed=0)
267 expected_step_finished = (sample_ids == end_token)
513 end_token = 6
524 end_fn = lambda sample_ids: math_ops.equal(sample_ids, end_token)
581 expected_step_finished = (sample_ids == end_token)
594 end_token =
[all...] |
| /external/tensorflow/tensorflow/contrib/seq2seq/kernels/ |
| beam_search_ops_gpu.cu.cc | 33 const T end_token, T* beams) {
70 beams[level_beam_ix] = end_token;
71 } else if (beams[level_beam_ix] == end_token) {
86 const T end_token, typename TTypes<T, 3>::Tensor beams) {
91 beams.device(d) = beams.constant(end_token);
101 end_token,
|
| beam_search_ops.h | 35 const T end_token, typename TTypes<T, 3>::Tensor beams);
|
| beam_search_ops.cc | 53 const Tensor& end_token = ctx->input(3); variable
64 ctx, TensorShapeUtils::IsScalar(end_token.shape()),
65 errors::InvalidArgument("end_token must be a scalar, saw shape: ",
66 end_token.shape().DebugString()));
87 typename TTypes<T>::ConstScalar end_token_t(end_token.scalar<T>());
111 const int32 end_token, TTypes<int32, 3>::Tensor beams) {
115 beams.setConstant(end_token);
117 auto DoWork = [&, ctx, end_token](int start_batch_beam,
147 beams(time, batch, beam) = end_token;
148 } else if (beams(time, batch, beam) == end_token) {
[all...] |
| /external/tensorflow/tensorflow/contrib/seq2seq/ops/ |
| beam_search_ops.cc | 29 .Input("end_token: T")
33 ShapeHandle step_ids, parent_ids, max_sequence_lengths, end_token;
37 // max_sequence_length is shaped [batch_size] and end_token is a scalar.
41 TF_RETURN_IF_ERROR(c->WithRank(c->input(3), 0, &end_token));
60 For a given beam, past the time step containing the first decoded `end_token`
61 all values are filled in with `end_token`.
68 end_token: `[]`.
|
| /external/tensorflow/tensorflow/contrib/seq2seq/python/ops/ |
| beam_search_decoder.py | 172 end_token,
184 end_token: `int32` scalar, the token that marks end of decoding.
196 `end_token` is not a scalar.
218 end_token, dtype=dtypes.int32, name="end_token")
220 raise ValueError("end_token must be a scalar")
344 end_token=self._end_token)
481 end_token = self._end_token
506 end_token=end_token,
[all...] |
| helper.py | 498 def __init__(self, embedding, start_tokens, end_token):
506 end_token: `int32` scalar, the token that marks end of decoding.
509 ValueError: if `start_tokens` is not a 1D tensor or `end_token` is not a
521 end_token, dtype=dtypes.int32, name="end_token")
526 raise ValueError("end_token must be a scalar")
575 def __init__(self, embedding, start_tokens, end_token,
584 end_token: `int32` scalar, the token that marks end of decoding.
593 ValueError: if `start_tokens` is not a 1D tensor or `end_token` is not a
597 embedding, start_tokens, end_token)
[all...] |
| /external/nanopb-c/generator/google/protobuf/ |
| text_format.py | 216 end_token = '>'
219 end_token = '}'
233 while not tokenizer.TryConsume(end_token):
235 raise tokenizer.ParseErrorPreviousToken('Expected "%s".' % (end_token))
|
| /external/protobuf/python/google/protobuf/ |
| text_format.py | 580 end_token = '>'
583 end_token = '}'
600 while not tokenizer.TryConsume(end_token):
602 raise tokenizer.ParseErrorPreviousToken('Expected "%s".' % (end_token,))
[all...] |
| /external/libtextclassifier/ |
| feature-processor.cc | 349 TokenIndex end_token = kInvalidIndex; local
363 end_token = i + 1;
366 return {start_token, end_token};
[all...] |
| /external/v8/src/parsing/ |
| parser-base.h | [all...] |
