Searched refs:kCellStateTensor (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | identify_lstm_merge_inputs.cc | 54 if (!GetMatchingRnnArray(model, src_op->outputs[kCellStateTensor], in Run()
76 src_op->outputs[kOutputTensor], src_op->outputs[kCellStateTensor])); in Run()
160 src_op->outputs[kCellStateTensor]; in Run()
|
| D | lstm_utils.h | 59 kCellStateTensor = 1, enumerator
|
| D | identify_lstm_split_inputs.cc | 157 lstm_cell_op->outputs[kCellStateTensor] = in Run()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | lstm_shared.h | 61 constexpr int kCellStateTensor = 19; variable
|
| D | unidirectional_sequence_lstm.cc | 58 GetVariableInput(context, node, lstm::full::kCellStateTensor); in PopulateQuantizedLstmParams8x8_16()
921 GetVariableInput(context, node, lstm::full::kCellStateTensor); in Prepare()
1282 GetVariableInput(context, node, lstm::full::kCellStateTensor); in Eval()
|
| D | lstm.cc | 150 GetVariableInput(context, node, kCellStateTensor); in PopulateQuantizedLstmParams8x8_16()
535 TfLiteTensor* cell_state = GetVariableInput(context, node, kCellStateTensor); in PopulateQuantizedLstmParams8x8_8()
1354 TfLiteTensor* cell_state = GetVariableInput(context, node, kCellStateTensor); in Prepare()
1898 TfLiteTensor* cell_state = GetVariableInput(context, node, kCellStateTensor); in Eval()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | lstm_parser.cc | 256 tflite::ops::builtin::lstm::full::kCellStateTensor, &cell_state)); in BuildCellStateUpdate()
454 tflite::ops::builtin::lstm::full::kCellStateTensor, &old_cell_state)); in ParseLSTMAttributes()
543 tflite::ops::builtin::lstm::full::kCellStateTensor, new_cell_state->id); in ParseLSTMAttributes()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| D | lstm.cc | 565 context, node, ops::builtin::lstm::full::kCellStateTensor); in lstm_eval()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
| D | lstm.cc | 598 context, node, ops::builtin::lstm::full::kCellStateTensor); in lstm_eval()
|