| /external/tensorflow/tensorflow/contrib/lite/kernels/ |
| lstm.cc | 255 TfLiteTensor* cell_state = GetOutput(context, node, kCellStateTensor); local
278 context->ResizeTensor(context, cell_state, cell_size));
282 cell_state->allocation_type = kTfLiteArenaRwPersistent;
349 TfLiteTensor* cell_state = GetOutput(context, node, kCellStateTensor); local
428 cell_to_input_weights->data.f, n_cell, cell_state->data.f, n_batch,
438 cell_to_forget_weights->data.f, n_cell, cell_state->data.f, n_batch,
446 cell_state->data.f, n_batch * n_cell,
447 cell_state->data.f);
455 cell_state->data.f);
458 cell_scratch, input_gate_scratch, n_batch * n_cell, cell_state->data.f)
[all...] |
| unidirectional_sequence_lstm.cc | 256 TfLiteTensor* cell_state = GetOutput(context, node, kCellStateTensor); local
280 context->ResizeTensor(context, cell_state, cell_size));
284 cell_state->allocation_type = kTfLiteArenaRwPersistent;
351 TfLiteTensor* cell_state = GetOutput(context, node, kCellStateTensor); local
436 cell_to_input_weights->data.f, n_cell, cell_state->data.f, n_batch,
446 cell_to_forget_weights->data.f, n_cell, cell_state->data.f, n_batch,
454 cell_state->data.f, n_batch * n_cell,
455 cell_state->data.f);
463 cell_state->data.f);
467 cell_state->data.f)
[all...] |
| /external/tensorflow/tensorflow/contrib/seq2seq/python/ops/ |
| basic_decoder.py | 138 cell_outputs, cell_state = self._cell(inputs, state)
142 time=time, outputs=cell_outputs, state=cell_state)
146 state=cell_state,
|
| attention_wrapper.py | [all...] |
| beam_search_decoder.py | 51 ("cell_state", "log_probs", "finished", "lengths"))):
144 `cell_state` value containing properly tiled final state from the
164 cell_state=tiled_encoder_final_state)
311 cell_state=self._initial_cell_state,
485 cell_state = state.cell_state
488 cell_state = nest.map_structure(self._maybe_merge_batch_beams, cell_state,
490 cell_outputs, next_cell_state = self._cell(inputs, cell_state)
630 # different gather_shape here because the cell_state tensors, i.e
[all...] |
| /frameworks/ml/nn/runtime/test/specs/V1_0/ |
| lstm.mod.py | 137 cell_state = [0, 0, 0, 0] variable
147 input0[cell_state_in] = cell_state
|
| lstm_state.mod.py | 137 cell_state = [-0.145439, 0.157475, 0.293663, -0.277353,] variable
147 input0[cell_state_in] = cell_state
|
| lstm_state2.mod.py | 137 cell_state = [-0.287121, 0.148115, 0.556837, -0.388276] variable
147 input0[cell_state_in] = cell_state
|
| /frameworks/ml/nn/runtime/test/specs/V1_1/ |
| lstm_relaxed.mod.py | 138 cell_state = [0, 0, 0, 0] variable
148 input0[cell_state_in] = cell_state
|
| lstm_state2_relaxed.mod.py | 138 cell_state = [-0.287121, 0.148115, 0.556837, -0.388276] variable
148 input0[cell_state_in] = cell_state
|
| lstm_state_relaxed.mod.py | 138 cell_state = [-0.145439, 0.157475, 0.293663, -0.277353,] variable
148 input0[cell_state_in] = cell_state
|
| /external/tensorflow/tensorflow/contrib/seq2seq/python/kernel_tests/ |
| attention_wrapper_test.py | 99 cell_state=state.cell_state, attention=state.attention)
101 self.assertEqual(state.cell_state.c.shape, static_state.cell_state.c.shape)
102 self.assertEqual(state.cell_state.h.shape, static_state.cell_state.h.shape)
196 isinstance(final_state.cell_state, rnn_cell.LSTMStateTuple))
206 tuple(final_state.cell_state.c.get_shape().as_list()))
208 tuple(final_state.cell_state.h.get_shape().as_list()))
269 cell_state=LSTMStateTuple
[all...] |
| beam_search_decoder_test.py | 120 cell_state=dummy_cell_state,
175 cell_state=dummy_cell_state,
276 cell_state=dummy_cell_state,
363 cell_state = cell.zero_state(
366 cell_state = cell_state.clone(cell_state=initial_state)
372 initial_state=cell_state,
|
| /external/tensorflow/tensorflow/contrib/grid_rnn/python/ops/ |
| grid_rnn_cell.py | 655 cell_state = (c_prev[i], last_dim_output)
658 cell_state = last_dim_output
665 new_output[d.idx], new_state[d.idx] = cell(cell_inputs, cell_state)
|
| /external/tensorflow/tensorflow/python/ops/ |
| rnn.py | [all...] |
| /external/tensorflow/tensorflow/contrib/rnn/python/kernel_tests/ |
| core_rnn_test.py | [all...] |
