| /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/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/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 | 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 …]
|
| 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
|
| 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
|
| /external/tensorflow/tensorflow/python/kernel_tests/nn_ops/ |
| 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
|
| /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 | 2512 const auto cell_state = op.input_cell_state().getType().cast<ShapedType>(); in verify() local
|