| /external/tensorflow/tensorflow/lite/experimental/examples/lstm/ |
| D | rnn.py | 44 sequence_length=None, argument
218 if sequence_length is not None:
219 sequence_length = math_ops.cast(sequence_length, dtypes.int32)
220 if sequence_length.shape.rank not in (None, 1):
223 "but saw shape: %s" % sequence_length.shape)
224 sequence_length = array_ops.identity( # Just to find it in the graph.
225 sequence_length,
250 if not context.executing_eagerly() and sequence_length is not None:
253 [_assert_has_shape(sequence_length, [batch_size])]):
254 sequence_length = array_ops.identity(
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | bidirectional_sequence_lstm_test.cc | 38 int sequence_length, bool use_cifg, in BidirectionalLSTMOpModel() argument
50 sequence_length_(sequence_length), in BidirectionalLSTMOpModel()
322 int sequence_length() { return sequence_length_; } in sequence_length() function in tflite::__anon0b0a63210111::BidirectionalLSTMOpModel
411 const int sequence_length = 3; in TEST_P() local
415 n_batch, n_input, n_cell, n_output, sequence_length, /*use_cifg=*/false, in TEST_P()
421 {sequence_length, n_batch, n_input}, // input tensor in TEST_P()
477 {sequence_length, n_batch, 0}, // aux_input tensor in TEST_P()
543 float* batch0_end = batch0_start + lstm.num_inputs() * lstm.sequence_length(); in TEST_P()
551 fw_golden_start + lstm.num_fw_outputs() * lstm.sequence_length(); in TEST_P()
560 bw_golden_start + lstm.num_bw_outputs() * lstm.sequence_length(); in TEST_P()
[all …]
|
| D | unidirectional_sequence_lstm_test.cc | 35 int sequence_length, bool time_major, bool use_cifg, in UnidirectionalLSTMOpModel() argument
45 sequence_length_(sequence_length) { in UnidirectionalLSTMOpModel()
201 int sequence_length() { return sequence_length_; } in sequence_length() function in tflite::__anon0e31a6420111::UnidirectionalLSTMOpModel
243 int n_batch, int n_input, int n_cell, int n_output, int sequence_length, in HybridUnidirectionalLSTMOpModel() argument
249 n_batch, n_input, n_cell, n_output, sequence_length, time_major, in HybridUnidirectionalLSTMOpModel()
456 const int sequence_length = 3; in TEST_F() local
459 n_batch, n_input, n_cell, n_output, sequence_length, in TEST_F()
465 {sequence_length, n_batch, n_input}, // input tensor in TEST_F()
518 const int sequence_length = 3; in TEST_F() local
521 n_batch, n_input, n_cell, n_output, sequence_length, in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/contrib/rnn/python/ops/ |
| D | fused_rnn_cell.py | 49 sequence_length=None, argument
99 sequence_length=None, argument
108 sequence_length=sequence_length,
124 sequence_length=sequence_length,
174 sequence_length=None, argument
176 inputs = self._reverse(inputs, sequence_length)
181 sequence_length=sequence_length,
183 outputs = self._reverse(outputs, sequence_length)
|
| D | rnn.py | 31 sequence_length=None, argument
116 sequence_length=sequence_length,
131 sequence_length=None, argument
235 sequence_length=sequence_length,
|
| D | lstm_ops.py | 461 sequence_length): argument
486 def call(self, inputs, initial_state=None, dtype=None, sequence_length=None): argument
541 if sequence_length is not None:
542 sequence_length = ops.convert_to_tensor(sequence_length)
545 inputs, initial_cell_state, initial_output, dtype, sequence_length)
547 if sequence_length is not None:
550 array_ops.sequence_mask(sequence_length, time_len, dtype=dtype),
563 final_cell_state = self._gather_states(mod_cell_states, sequence_length,
565 final_output = self._gather_states(mod_outputs, sequence_length,
658 sequence_length=None): argument
[all …]
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | rnn.py | 173 time, sequence_length, min_sequence_length, max_sequence_length, argument
231 copy_cond = time >= sequence_length
355 def bidirectional_dynamic_rnn(cell_fw, cell_bw, inputs, sequence_length=None, argument
440 cell=cell_fw, inputs=inputs, sequence_length=sequence_length,
466 seq_lengths=sequence_length,
472 cell=cell_bw, inputs=inputs_reverse, sequence_length=sequence_length,
478 tmp, seq_lengths=sequence_length,
491 def dynamic_rnn(cell, inputs, sequence_length=None, initial_state=None, argument
625 if sequence_length is not None:
626 sequence_length = math_ops.cast(sequence_length, dtypes.int32)
[all …]
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
| D | state_saving_rnn_estimator_test.py | 248 sequence_length = 4
263 1.0, shape=[sequence_length]),
268 labels = constant_op.constant(5.0, shape=[sequence_length])
383 sequence_length = 10
397 [sequence_length + 1], 0, 2, dtype=dtypes.int32, seed=seed)
398 labels = array_ops.slice(random_sequence, [0], [sequence_length])
400 array_ops.slice(random_sequence, [1], [sequence_length]),
448 sequence_length, argument
452 [sequence_length + 1], 0, 2, dtype=dtypes.int32, seed=seed)
453 labels = array_ops.slice(random_sequence, [0], [sequence_length])
[all …]
|
| D | dynamic_rnn_estimator_test.py | 358 sequence_length = 16
363 def get_shift_input_fn(batch_size, sequence_length, seed=None): argument
367 [batch_size, sequence_length + 1],
373 [batch_size, sequence_length])
377 [batch_size, sequence_length]),
403 train_input_fn = get_shift_input_fn(batch_size, sequence_length, seed=12321)
404 eval_input_fn = get_shift_input_fn(batch_size, sequence_length, seed=32123)
424 def get_input_fn(batch_size, sequence_length, state_dict, starting_step=0): argument
428 [[(starting_step + i + j) % 2 for j in range(sequence_length + 1)]
432 [batch_size, sequence_length])
[all …]
|
| D | rnn_common_test.py | 38 sequence_length = np.random.randint(0, padded_length + 1, batch_size)
45 constant_op.constant(sequence_length, dtype=dtypes.int32))
51 expected_activations_shape = [sum(sequence_length), num_classes]
57 expected_labels_shape = [sum(sequence_length)]
63 for j in range(sequence_length[i]):
89 sequence_length = np.random.randint(0, padded_length + 1, batch_size)
93 constant_op.constant(sequence_length, dtype=dtypes.int32))
106 expected_activations = activations[i, sequence_length[i] - 1, :]
|
| D | dynamic_rnn_estimator.py | 261 sequence_length, argument
290 activations, sequence_length)
308 activations, labels, sequence_length, target_column, features): argument
326 activations, labels, sequence_length)
331 activations, labels, sequence_length, target_column, features): argument
350 activations, sequence_length)
482 sequence_length = features.get(sequence_length_key)
507 rnn_activations, labels, sequence_length, target_column, features)
510 rnn_activations, sequence_length, target_column,
514 rnn_activations, labels, sequence_length, target_column, features)
[all …]
|
| /external/v8/src/ |
| D | layout-descriptor.cc | 142 int sequence_length; in IsTagged() local
144 sequence_length = base::bits::CountTrailingZeros(value) - layout_bit_index; in IsTagged()
146 if (layout_bit_index + sequence_length == kBitsPerLayoutWord) { in IsTagged()
157 sequence_length += cur_sequence_length; in IsTagged()
158 if (sequence_length >= max_sequence_length) break; in IsTagged()
161 if (is_tagged && (field_index + sequence_length == capacity())) { in IsTagged()
165 sequence_length = std::numeric_limits<int>::max(); in IsTagged()
169 sequence_length = Min(base::bits::CountTrailingZeros(value), in IsTagged()
172 if (is_tagged && (field_index + sequence_length == capacity())) { in IsTagged()
176 sequence_length = std::numeric_limits<int>::max(); in IsTagged()
[all …]
|
| D | collector.h | 205 int sequence_length = this->index_ - sequence_start_; in DropSequence() local
207 this->size_ -= sequence_length; in DropSequence()
227 int sequence_length = this->index_ - sequence_start_; in NewChunk() local
228 Vector<T> new_chunk = Vector<T>::New(sequence_length + new_capacity); in NewChunk()
229 DCHECK(sequence_length < new_chunk.length()); in NewChunk()
230 for (int i = 0; i < sequence_length; i++) { in NewChunk()
240 this->index_ = sequence_length; in NewChunk()
|
| /external/tensorflow/tensorflow/contrib/seq2seq/python/kernel_tests/ |
| D | decoder_v2_test.py | 40 sequence_length = [3, 4, 3, 1, 0]
45 max_out = max(sequence_length)
66 input_t, initial_state=initial_state, sequence_length=sequence_length)
89 expected_length = sequence_length
117 sequence_length = [3, 4, 3, 1, 0]
122 max_out = max(sequence_length)
136 inputs, initial_state=zero_state, sequence_length=sequence_length)
141 sequence_length=sequence_length if use_sequence_length else None,
|
| D | loss_test.py | 36 self.sequence_length = 3
41 for i in range(self.sequence_length)
46 for i in range(self.sequence_length)
51 for _ in range(self.sequence_length)
73 compare_per_sequence = np.full((self.sequence_length), self.expected_loss)
89 compare_total = np.full((self.batch_size, self.sequence_length),
112 compare_per_sequence = np.full((self.sequence_length), self.expected_loss)
132 compare_total = np.full((self.batch_size, self.sequence_length),
155 compare_per_sequence = np.full((self.sequence_length), self.expected_loss)
175 compare_total = np.full((self.batch_size, self.sequence_length),
[all …]
|
| D | decoder_test.py | 39 sequence_length = [3, 4, 3, 1, 0]
44 max_out = max(sequence_length)
55 inputs, sequence_length, time_major=time_major)
94 expected_length = sequence_length
121 sequence_length = [3, 4, 3, 1, 0]
126 max_out = max(sequence_length)
134 helper = helper_py.TrainingHelper(inputs, sequence_length)
152 sequence_length=sequence_length if use_sequence_length else None,
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ctc_ops.cc | 41 ShapeHandle sequence_length; in __anon78c05cd60102() local
46 TF_RETURN_IF_ERROR(c->WithRank(c->input(3), 1, &sequence_length)); in __anon78c05cd60102()
56 c->Merge(c->Dim(inputs, 1), c->Dim(sequence_length, 0), &batch_size)); in __anon78c05cd60102()
74 ShapeHandle sequence_length; in __anon78c05cd60202() local
77 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 1, &sequence_length)); in __anon78c05cd60202()
82 c->Merge(c->Dim(inputs, 1), c->Dim(sequence_length, 0), &batch_size)); in __anon78c05cd60202()
104 ShapeHandle sequence_length; in __anon78c05cd60302() local
107 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 1, &sequence_length)); in __anon78c05cd60302()
112 c->Merge(c->Dim(inputs, 1), c->Dim(sequence_length, 0), &batch_size)); in __anon78c05cd60302()
|
| /external/tensorflow/tensorflow/contrib/rnn/python/kernel_tests/ |
| D | rnn_test.py | 57 sequence_length = array_ops.placeholder(
87 sequence_length=sequence_length,
99 return input_value, inputs, outputs, state_fw, state_bw, sequence_length
103 input_value, inputs, outputs, state_fw, state_bw, sequence_length = (
110 sequence_length: [2, 3]})
176 input_value, inputs, outputs, state_fw, state_bw, sequence_length = (
183 feed_dict = {inputs[0]: input_value, sequence_length: [3, 2]}
194 feed_dict = {inputs[0]: input_value, sequence_length: [5, 3]}
204 feed_dict = {inputs[0]: input_value, sequence_length: [2, 1]}
237 sequence_length = array_ops.placeholder(dtypes.int64)
[all …]
|
| /external/tensorflow/tensorflow/contrib/recurrent/python/ops/ |
| D | functional_rnn.py | 198 def _PickFinalStateFromHistory(acc_state, sequence_length): argument
209 lengths = array_ops.tile(array_ops.reshape(sequence_length,
277 sequence_length=None, argument
294 if sequence_length is not None:
295 max_length = math_ops.reduce_max(sequence_length)
314 sequence_length,
328 sequence_length=None, argument
392 cell=cell_fw, inputs=inputs, sequence_length=sequence_length,
418 seq_lengths=sequence_length,
424 sequence_length=sequence_length,
[all …]
|
| /external/tensorflow/tensorflow/contrib/feature_column/python/feature_column/ |
| D | sequence_feature_column_test.py | 124 input_layer, sequence_length = sfc.sequence_input_layer(
142 expected_sequence_length, sequence_length.eval(session=sess))
216 input_layer, sequence_length = sfc.sequence_input_layer(
231 expected_sequence_length, sequence_length.eval(session=sess))
324 input_layer, sequence_length = sfc.sequence_input_layer(
335 expected_sequence_length, sequence_length.eval(session=sess))
389 input_layer, sequence_length = sfc.sequence_input_layer(
396 expected_sequence_length, sequence_length.eval(session=sess))
434 input_layer, sequence_length = sfc.sequence_input_layer(
441 expected_sequence_length, sequence_length.eval(session=sess))
[all …]
|
| /external/tensorflow/tensorflow/python/feature_column/ |
| D | sequence_feature_column_test.py | 130 input_layer, sequence_length = sequence_input_layer({
143 expected_sequence_length, self.evaluate(sequence_length))
221 input_layer, sequence_length = sequence_input_layer({
234 expected_sequence_length, sequence_length.eval(session=sess))
329 input_layer, sequence_length = sequence_input_layer({
334 expected_sequence_length, self.evaluate(sequence_length))
390 input_layer, sequence_length = sequence_input_layer({'aaa': sparse_input})
394 expected_sequence_length, self.evaluate(sequence_length))
434 input_layer, sequence_length = sequence_input_layer({'aaa': sparse_input})
438 expected_sequence_length, self.evaluate(sequence_length))
[all …]
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/ |
| D | models.py | 223 sequence_length=None, argument
277 sequence_length)
283 _reverse_seq(inputs, sequence_length), initial_state_bw, dtype,
284 sequence_length)
285 output_bw = _reverse_seq(tmp, sequence_length)
299 target_predictor_fn, sequence_length, initial_state, argument
380 sequence_length=sequence_length,
403 sequence_length=sequence_length,
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | rnn_test.py | 172 sequence_length=[[4]])
246 cell, inputs, dtype=dtypes.float32, sequence_length=[4])
266 cell, inputs, dtype=dtypes.float32, sequence_length=[4])
281 rnn.dynamic_rnn(cell, inputs, dtype=dtypes.float32, sequence_length=[4])
296 cell, inputs, dtype=dtypes.float32, sequence_length=[4])
718 def _static_vs_dynamic_rnn_benchmark_static(inputs_list_t, sequence_length): argument
729 sequence_length=sequence_length,
740 def _static_vs_dynamic_rnn_benchmark_dynamic(inputs_t, sequence_length): argument
749 cell, inputs_t, sequence_length=sequence_length, dtype=dtypes.float32)
769 sequence_length = np.random.randint(0, max_time, size=batch_size)
[all …]
|
| D | rnn_cell_test.py | 206 rnn.static_rnn(cell, inputs, dtype=dtypes.float32, sequence_length=4)
276 sequence_length = array_ops.placeholder(dtypes.int64)
285 cell, inputs, sequence_length=sequence_length, dtype=dtypes.float32)
294 sequence_length: [2, 3]
300 sequence_length: [2, 3]
608 sequence_length = [4, 6]
632 sequence_length=sequence_length,
646 sequence_length=sequence_length,
810 sequence_length = array_ops.placeholder(dtypes.int64)
830 sequence_length=sequence_length,
[all …]
|
| /external/tensorflow/tensorflow/contrib/seq2seq/python/ops/ |
| D | sampler.py | 219 def initialize(self, inputs, sequence_length=None): argument
237 if sequence_length is None:
239 self.sequence_length = ops.convert_to_tensor(
240 sequence_length, name="sequence_length")
241 if self.sequence_length.get_shape().ndims != 1:
249 self._batch_size = array_ops.size(self.sequence_length)
251 finished = math_ops.equal(0, self.sequence_length)
267 finished = (next_time >= self.sequence_length)
325 def initialize(self, inputs, sequence_length=None, embedding=None): argument
333 inputs, sequence_length=sequence_length)
[all …]
|