| /external/tensorflow/tensorflow/contrib/seq2seq/python/kernel_tests/ |
| D | 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 = 5
108 0, high=end_token + 1, size=(max_time, batch_size, beam_width))
116 end_token=end_token)
123 self.assertAllClose(b_value, end_token * np.ones_like(b_value))
[all …]
|
| D | 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 - 1
371 end_token=end_token,
|
| D | 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 = 6
[all …]
|
| /external/tensorflow/tensorflow/contrib/seq2seq/kernels/ |
| D | beam_search_ops.cc | 53 const Tensor& end_token = ctx->input(3); in Compute() local
64 ctx, TensorShapeUtils::IsScalar(end_token.shape()), in Compute()
66 end_token.shape().DebugString())); in Compute()
87 typename TTypes<T>::ConstScalar end_token_t(end_token.scalar<T>()); in Compute()
111 const int32 end_token, TTypes<int32, 3>::Tensor beams) { in operator ()()
115 beams.setConstant(end_token); in operator ()()
117 auto DoWork = [&, ctx, end_token](int start_batch_beam, in operator ()()
147 beams(time, batch, beam) = end_token; in operator ()()
148 } else if (beams(time, batch, beam) == end_token) { in operator ()()
176 TTypes<int32>::ConstVec max_sequence_lengths, const T end_token, \
|
| D | beam_search_ops_gpu.cu.cc | 33 const T end_token, T* beams) { in GatherTreeOpKernel() argument
70 beams[level_beam_ix] = end_token; in GatherTreeOpKernel()
71 } else if (beams[level_beam_ix] == end_token) { in GatherTreeOpKernel()
86 const T end_token, typename TTypes<T, 3>::Tensor beams) { in operator ()()
91 beams.device(d) = beams.constant(end_token); in operator ()()
101 end_token, in operator ()()
|
| D | beam_search_ops.h | 35 const T end_token, typename TTypes<T, 3>::Tensor beams);
|
| /external/tensorflow/tensorflow/contrib/seq2seq/python/ops/ |
| D | beam_search_decoder.py | 172 end_token, argument
218 end_token, dtype=dtypes.int32, name="end_token")
344 end_token=self._end_token)
481 end_token = self._end_token
506 end_token=end_token,
519 beam_width, end_token, length_penalty_weight): argument
549 step_log_probs = _mask_probs(step_log_probs, end_token, previously_finished)
555 indices=array_ops.fill([batch_size, beam_width], end_token),
612 math_ops.equal(next_word_ids, end_token),
|
| D | helper.py | 498 def __init__(self, embedding, start_tokens, end_token): argument
521 end_token, dtype=dtypes.int32, name="end_token")
575 def __init__(self, embedding, start_tokens, end_token, argument
597 embedding, start_tokens, end_token)
|
| /external/tensorflow/tensorflow/contrib/seq2seq/ops/ |
| D | beam_search_ops.cc | 33 ShapeHandle step_ids, parent_ids, max_sequence_lengths, end_token; in __anon4dc23df50102() local
41 TF_RETURN_IF_ERROR(c->WithRank(c->input(3), 0, &end_token)); in __anon4dc23df50102()
|
| /external/nanopb-c/generator/google/protobuf/ |
| D | 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/ |
| D | text_format.py | 580 end_token = '>'
583 end_token = '}'
600 while not tokenizer.TryConsume(end_token):
602 raise tokenizer.ParseErrorPreviousToken('Expected "%s".' % (end_token,))
|
| /external/libtextclassifier/ |
| D | feature-processor.cc | 349 TokenIndex end_token = kInvalidIndex; in CodepointSpanToTokenSpan() local
363 end_token = i + 1; in CodepointSpanToTokenSpan()
366 return {start_token, end_token}; in CodepointSpanToTokenSpan()
|
| /external/tensorflow/tensorflow/docs_src/api_guides/python/ |
| D | contrib.seq2seq.md | 107 end_token=END_SYMBOL)
|
| /external/v8/src/parsing/ |
| D | parser-base.h | 1213 V8_INLINE void ParseStatementList(StatementListT body, int end_token, in ParseStatementList() argument
1215 LazyParsingResult result = ParseStatementList(body, end_token, false, ok); in ParseStatementList()
1219 LazyParsingResult ParseStatementList(StatementListT body, int end_token,
4604 ParserBase<Impl>::ParseStatementList(StatementListT body, int end_token, in ParseStatementList() argument
4619 while (peek() != end_token) { in ParseStatementList()
|