| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| D | lstm.cc | 44 const float* cell_state, const float* cell_to_gate_weights, in CalculateLstmGateFloat() argument
81 cell_to_gate_weights, n_cell, cell_state, n_batch, gate); in CalculateLstmGateFloat()
100 void UpdateLstmCellFloat(int n_batch, int n_cell, float* cell_state, in UpdateLstmCellFloat() argument
103 tensor_utils::VectorVectorCwiseProduct(forget_gate, cell_state, in UpdateLstmCellFloat()
104 n_batch * n_cell, cell_state); in UpdateLstmCellFloat()
113 cell_gate, scratch, n_batch * n_cell, cell_state); in UpdateLstmCellFloat()
116 cell_gate, input_gate, n_batch * n_cell, cell_state); in UpdateLstmCellFloat()
119 tensor_utils::CwiseClipping(cell_state, n_batch * n_cell, clip); in UpdateLstmCellFloat()
124 int n_batch, int n_cell, int n_output, const float* cell_state, in CalculateLstmOutputCalibration() argument
129 tensor_utils::ApplyActivationToVector(cell_state, n_batch * n_cell, in CalculateLstmOutputCalibration()
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | lstm_eval.cc | 154 const float* cell_state, const float* cell_to_gate_weights, in CalculateLstmGateFloat() argument
189 cell_to_gate_weights, n_cell, cell_state, n_batch, gate); in CalculateLstmGateFloat()
218 void UpdateLstmCellFloat(int n_batch, int n_cell, float* cell_state, in UpdateLstmCellFloat() argument
221 tensor_utils::VectorVectorCwiseProduct(forget_gate, cell_state, in UpdateLstmCellFloat()
222 n_batch * n_cell, cell_state); in UpdateLstmCellFloat()
231 cell_gate, scratch, n_batch * n_cell, cell_state); in UpdateLstmCellFloat()
234 cell_gate, input_gate, n_batch * n_cell, cell_state); in UpdateLstmCellFloat()
237 tensor_utils::CwiseClipping(cell_state, n_batch * n_cell, clip); in UpdateLstmCellFloat()
263 const float* cell_state, const float* output_gate, in CalculateLstmOutputFloat() argument
269 tensor_utils::ApplyActivationToVector(cell_state, n_batch * n_cell, in CalculateLstmOutputFloat()
[all …]
|
| D | lstm_eval.h | 124 TfLiteTensor* cell_state, TfLiteTensor* output);
166 TfLiteTensor* output_state, TfLiteTensor* cell_state,
193 TfLiteTensor* output_state, TfLiteTensor* cell_state, TfLiteTensor* output,
218 TfLiteTensor* cell_state, TfLiteTensor* output,
|
| D | unidirectional_sequence_lstm.cc | 57 const TfLiteTensor* cell_state = in PopulateQuantizedLstmParams8x8_16() local
59 TF_LITE_ENSURE(context, cell_state != nullptr); in PopulateQuantizedLstmParams8x8_16()
66 static_cast<TfLiteAffineQuantization*>(cell_state->quantization.params); in PopulateQuantizedLstmParams8x8_16()
255 TF_LITE_ENSURE(context, CheckedLog2(cell_state->params.scale, &cell_scale)); in PopulateQuantizedLstmParams8x8_16()
920 TfLiteTensor* cell_state = in Prepare() local
922 TF_LITE_ENSURE(context, cell_state != nullptr); in Prepare()
928 TF_LITE_ENSURE_EQ(context, NumElements(cell_state), n_batch * n_cell); in Prepare()
1014 if (!TfLiteIntArrayEqual(cell_state_quantized->dims, cell_state->dims)) { in Prepare()
1016 TfLiteIntArrayCopy(cell_state->dims); in Prepare()
1281 TfLiteTensor* cell_state = in Eval() local
[all …]
|
| D | lstm.cc | 149 const TfLiteTensor* cell_state = in PopulateQuantizedLstmParams8x8_16() local
151 TF_LITE_ENSURE(context, cell_state != nullptr); in PopulateQuantizedLstmParams8x8_16()
157 static_cast<TfLiteAffineQuantization*>(cell_state->quantization.params); in PopulateQuantizedLstmParams8x8_16()
340 TF_LITE_ENSURE(context, CheckedLog2(cell_state->params.scale, &cell_scale)); in PopulateQuantizedLstmParams8x8_16()
535 TfLiteTensor* cell_state = GetVariableInput(context, node, kCellStateTensor); in PopulateQuantizedLstmParams8x8_8() local
536 TF_LITE_ENSURE(context, cell_state != nullptr); in PopulateQuantizedLstmParams8x8_8()
667 cell_ptr = cell_state->data.i16; in PopulateQuantizedLstmParams8x8_8()
816 cell_state->quantization.params); in PopulateQuantizedLstmParams8x8_8()
1354 TfLiteTensor* cell_state = GetVariableInput(context, node, kCellStateTensor); in Prepare() local
1355 TF_LITE_ENSURE(context, cell_state != nullptr); in Prepare()
[all …]
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | lstm_parser.cc | 116 Value* output_state, Value* cell_state, in BuildLstmGate() argument
123 Value* input_times_weights = CreateNewSimilarValue(graph, cell_state); in BuildLstmGate()
141 Value* output_state_times_weights = CreateNewSimilarValue(graph, cell_state); in BuildLstmGate()
158 cell_state_times_weights = CreateNewSimilarValue(graph, cell_state); in BuildLstmGate()
166 RETURN_IF_ERROR(graph->AddConsumer(node->id, cell_state->id)); in BuildLstmGate()
170 Value* gate_before_normalization = CreateNewSimilarValue(graph, cell_state); in BuildLstmGate()
254 Value* cell_state; in BuildCellStateUpdate() local
256 tflite::ops::builtin::lstm::full::kCellStateTensor, &cell_state)); in BuildCellStateUpdate()
263 RETURN_IF_ERROR(graph->AddConsumer(node->id, cell_state->id)); in BuildCellStateUpdate()
338 Value* cell_state, in BuildOutputStateUpdate() argument
[all …]
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
| D | lstm.cc | 357 TfLiteTensor* activation_state, TfLiteTensor* cell_state, in EvalFloat()
445 GetTensorData<float>(cell_state), input_gate_scratch, in EvalFloat()
471 float* cell_state_ptr = GetTensorData<float>(cell_state) + b * n_cell; in EvalFloat()
597 TfLiteTensor* cell_state = GetVariableInput( in lstm_eval() local
599 TF_LITE_ENSURE(context, cell_state != nullptr); in lstm_eval()
627 /*output_offset=*/0, scratch_buffer, activation_state, cell_state, in lstm_eval()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | rnn.py | 1180 (next_output, cell_state) = cell(current_input, state)
1182 nest.assert_same_structure(state, cell_state)
1187 next_loop_state) = loop_fn(next_time, next_output, cell_state,
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | rnn_cell_test.py | 2076 def loop_fn(time_, cell_output, cell_state, unused_loop_state): argument
2081 next_state = cell_state # copy state through
2185 def loop_fn(time_, cell_output, cell_state, loop_state): argument
2191 next_state = cell_state
2222 def loop_fn(time_, cell_output, cell_state, loop_state): argument
2234 next_state = cell_state
2269 def loop_fn(time_, cell_output, cell_state, _): argument
2278 next_state = cell_state
2338 def loop_fn(time_, cell_output, cell_state, unused_loop_state): argument
2343 next_state = cell_state
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.cc | 2019 const auto cell_state = op.input_cell_state().getType().cast<ShapedType>(); in Verify() local
2024 if (activation_state.hasStaticShape() && cell_state.hasStaticShape() && in Verify()
2033 const int state_size = cell_state.getNumElements(); in Verify()
|