| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | lstm_parser.cc | 116 Value* output_state, Value* cell_state, in BuildLstmGate()
254 Value* cell_state; in BuildCellStateUpdate() local
338 Value* cell_state, in BuildOutputStateUpdate()
|
| /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()
263 TfLiteStatus EvalCalibration( in EvalCalibration()
564 TfLiteTensor* cell_state = GetVariableInput( in lstm_eval() local
|
| /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
920 TfLiteTensor* cell_state = in Prepare() local
1281 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 | 331 TfLiteStatus EvalFloat( in EvalFloat()
597 TfLiteTensor* cell_state = GetVariableInput( in lstm_eval() local
|
| /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
|