| /external/ComputeLibrary/tests/validation/NEON/ |
| 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
|
| /external/ComputeLibrary/tests/validation/CL/ |
| 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
|
| /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()
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
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| 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()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | lstm_eval.cc | 154 const float* cell_state, const float* cell_to_gate_weights, in CalculateLstmGateFloat()
218 void UpdateLstmCellFloat(int n_batch, int n_cell, float* cell_state, in UpdateLstmCellFloat()
263 const float* cell_state, const float* output_gate, in CalculateLstmOutputFloat()
316 const float* cell_state, const int8_t* cell_to_gate_weights, in CalculateLstmGateHybrid()
433 int n_batch, int n_cell, int n_output, const float* cell_state, in CalculateLstmOutputHybrid()
498 const int16_t* cell_state, const int16_t* cell_to_gate_weights, in CalculateLstmGateInteger8x8_16()
571 void UpdateLstmCellInteger(int n_batch, int n_cell, int16_t* cell_state, in UpdateLstmCellInteger()
617 int n_batch, int n_cell, int n_output, const int16_t* cell_state, in CalculateLstmOutputInteger8x8_16()
725 int n_batch, int n_cell, int n_output, const int16_t* cell_state, in CalculateLstmOutputInteger8x8_8()
1687 TfLiteStatus EvalFloat( in EvalFloat()
[all …]
|
| D | unidirectional_sequence_lstm.cc | 57 const TfLiteTensor* cell_state = in PopulateQuantizedLstmParams8x8_16() local
929 TfLiteTensor* cell_state = in Prepare() local
1290 TfLiteTensor* cell_state = in Eval() local
|
| D | lstm.cc | 149 const TfLiteTensor* cell_state = in PopulateQuantizedLstmParams8x8_16() local
535 TfLiteTensor* cell_state = GetVariableInput(context, node, kCellStateTensor); in PopulateQuantizedLstmParams8x8_8() local
1354 TfLiteTensor* cell_state = GetVariableInput(context, node, kCellStateTensor); in Prepare() local
1898 TfLiteTensor* cell_state = GetVariableInput(context, node, kCellStateTensor); in Eval() local
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | rnn_cell_test.py | 2076 def loop_fn(time_, cell_output, cell_state, unused_loop_state): argument
2185 def loop_fn(time_, cell_output, cell_state, loop_state): argument
2222 def loop_fn(time_, cell_output, cell_state, loop_state): argument
2269 def loop_fn(time_, cell_output, cell_state, _): argument
2338 def loop_fn(time_, cell_output, cell_state, unused_loop_state): argument
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
| D | lstm.cc | 340 TfLiteStatus EvalFloat( in EvalFloat()
609 TfLiteTensor* cell_state = GetVariableInput( in lstm_eval() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.cc | 2331 const auto cell_state = op.input_cell_state().getType().cast<ShapedType>(); in Verify() local
|