| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | lstm_eval.cc | 185 const float* cell_state, const float* cell_to_gate_weights, in CalculateLstmGateFloat()
255 void UpdateLstmCellFloat(int n_batch, int n_cell, float* cell_state, in UpdateLstmCellFloat()
303 const float* cell_state, const float* output_gate, in CalculateLstmOutputFloat()
358 const float* cell_state, const int8_t* cell_to_gate_weights, in CalculateLstmGateHybrid()
475 int n_batch, int n_cell, int n_output, const float* cell_state, in CalculateLstmOutputHybrid()
540 const int16_t* cell_state, const int16_t* cell_to_gate_weights, in CalculateLstmGateInteger8x8_16()
613 void UpdateLstmCellInteger(int n_batch, int n_cell, int16_t* cell_state, in UpdateLstmCellInteger()
659 int n_batch, int n_cell, int n_output, const int16_t* cell_state, in CalculateLstmOutputInteger8x8_16()
767 int n_batch, int n_cell, int n_output, const int16_t* cell_state, in CalculateLstmOutputInteger8x8_8()
1745 TfLiteStatus EvalFloat( in EvalFloat()
[all …]
|
| H A D | unidirectional_sequence_lstm.cc | 58 const TfLiteTensor* cell_state = in PopulateQuantizedLstmParams8x8_16() local
930 TfLiteTensor* cell_state = in Prepare() local
1294 TfLiteTensor* cell_state = in Eval() local
|
| H A D | lstm.cc | 150 const TfLiteTensor* cell_state = in PopulateQuantizedLstmParams8x8_16() local
536 TfLiteTensor* cell_state = GetVariableInput(context, node, kCellStateTensor); in PopulateQuantizedLstmParams8x8_8() local
1355 TfLiteTensor* cell_state = GetVariableInput(context, node, kCellStateTensor); in Prepare() local
1902 TfLiteTensor* cell_state = GetVariableInput(context, node, kCellStateTensor); in Eval() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| H A D | lstm.cc | 44 const float* cell_state, const float* cell_to_gate_weights, in CalculateLstmGateFloat()
100 void UpdateLstmCellFloat(int n_batch, int n_cell, float* cell_state, in UpdateLstmCellFloat()
124 int n_batch, int n_cell, int n_output, const float* cell_state, in CalculateLstmOutputCalibration()
264 TfLiteStatus EvalCalibration( in EvalCalibration()
565 TfLiteTensor* cell_state = GetVariableInput( in lstm_eval() local
|
| /aosp_15_r20/external/ComputeLibrary/tests/validation/CL/ |
| H A D | LSTMLayerQuantized.cpp | 113 auto cell_state = create_tensor<CLTensor>(output_shape, DataType::QSYMM16, 1, qsymm_4); in TEST_SUITE() local
252 auto cell_state = create_tensor<CLTensor>(output_shape, DataType::QSYMM16, 1, qsymm_4); in TEST_CASE() local
492 auto cell_state = create_tensor<CLTensor>(output_shape, DataType::QSYMM16, 1, qsymm_4); in TEST_SUITE() local
|
| /aosp_15_r20/external/ComputeLibrary/tests/validation/NEON/ |
| H A D | LSTMLayerQuantized.cpp | 120 auto cell_state = create_tensor<Tensor>(output_shape, DataType::QSYMM16, 1, qsymm_4); in TEST_SUITE() local
259 auto cell_state = create_tensor<Tensor>(output_shape, DataType::QSYMM16, 1, qsymm_4); in TEST_CASE() local
499 auto cell_state = create_tensor<Tensor>(output_shape, DataType::QSYMM16, 1, qsymm_4); in TEST_SUITE() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| H A D | lstm_parser.cc | 148 Value* output_state, Value* cell_state, in BuildLstmGate()
278 Value* cell_state; in BuildCellStateUpdate() local
362 Value* cell_state, in BuildOutputStateUpdate()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/nn_ops/ |
| H A D | rnn_cell_test.py | 2154 def loop_fn(time_, cell_output, cell_state, unused_loop_state): argument
2263 def loop_fn(time_, cell_output, cell_state, loop_state): argument
2300 def loop_fn(time_, cell_output, cell_state, loop_state): argument
2347 def loop_fn(time_, cell_output, cell_state, _): argument
2416 def loop_fn(time_, cell_output, cell_state, unused_loop_state): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
| H A D | lstm.cc | 340 TfLiteStatus EvalFloat( in EvalFloat()
609 TfLiteTensor* cell_state = GetVariableInput( in lstm_eval() local
|
| /aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/specs/V1_0/ |
| D | lstm_state.mod.py | 137 cell_state = [-0.145439, 0.157475, 0.293663, -0.277353,] variable
|
| D | lstm_state2.mod.py | 137 cell_state = [-0.287121, 0.148115, 0.556837, -0.388276] variable
|
| D | lstm.mod.py | 137 cell_state = [0, 0, 0, 0] variable
|
| /aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/specs/V1_2/ |
| D | lstm_state_float16.mod.py | 137 cell_state = [-0.145439, 0.157475, 0.293663, -0.277353,] variable
|
| D | lstm_state2_float16.mod.py | 137 cell_state = [-0.287121, 0.148115, 0.556837, -0.388276] variable
|
| D | lstm_float16.mod.py | 137 cell_state = [0, 0, 0, 0] variable
|
| /aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/specs/V1_1/ |
| D | lstm_relaxed.mod.py | 138 cell_state = [0, 0, 0, 0] variable
|
| D | lstm_state_relaxed.mod.py | 138 cell_state = [-0.145439, 0.157475, 0.293663, -0.277353,] variable
|
| D | lstm_state2_relaxed.mod.py | 138 cell_state = [-0.287121, 0.148115, 0.556837, -0.388276] variable
|
| /aosp_15_r20/packages/modules/NeuralNetworks/tools/test_generator/tests/P_backward_compatibility_float/ |
| D | lstm_float.mod.py | 134 cell_state = [0, 0, 0, 0] variable
|
| /aosp_15_r20/packages/modules/NeuralNetworks/tools/test_generator/tests/P_vts_backward_compatibility_float/ |
| D | lstm_float.mod.py | 134 cell_state = [0, 0, 0, 0] variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| H A D | tfl_ops.cc | 2512 const auto cell_state = op.input_cell_state().getType().cast<ShapedType>(); in verify() local
|