Searched refs:kOutputStateTensor (Results 1 – 9 of 9) sorted by relevance
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
H A D | lstm_shared.h | 60 constexpr int kOutputStateTensor = 18; variable
|
H A D | unidirectional_sequence_lstm.cc | 151 GetVariableInput(context, node, lstm::full::kOutputStateTensor); in PopulateQuantizedLstmParams8x8_16() 720 GetVariableInput(context, node, lstm::full::kOutputStateTensor); in PopulatePrecomputedZPTimesWeightsWithBias() 928 GetVariableInput(context, node, lstm::full::kOutputStateTensor); in Prepare() 1292 GetVariableInput(context, node, lstm::full::kOutputStateTensor); in Eval()
|
H A D | lstm.cc | 232 GetVariableInput(context, node, kOutputStateTensor); in PopulateQuantizedLstmParams8x8_16() 534 GetVariableInput(context, node, kOutputStateTensor); in PopulateQuantizedLstmParams8x8_8() 1159 GetVariableInput(context, node, kOutputStateTensor); in PopulatePrecomputedZPTimesWeightsWithBias() 1353 GetVariableInput(context, node, kOutputStateTensor); in Prepare() 1900 GetVariableInput(context, node, kOutputStateTensor); in Eval()
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
H A D | lstm_utils.h | 58 kOutputStateTensor = 0, enumerator
|
H A D | identify_lstm_split_inputs.cc | 157 lstm_cell_op->outputs[kOutputStateTensor] = in Run()
|
H A D | identify_lstm_merge_inputs.cc | 164 src_op->outputs[kOutputStateTensor]; in Run()
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
H A D | lstm_parser.cc | 481 tflite::ops::builtin::lstm::full::kOutputStateTensor, &old_output_state)); in ParseLSTMAttributes() 563 tflite::ops::builtin::lstm::full::kOutputStateTensor, in ParseLSTMAttributes()
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
H A D | lstm.cc | 607 context, node, ops::builtin::lstm::full::kOutputStateTensor); in lstm_eval()
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
H A D | lstm.cc | 563 context, node, ops::builtin::lstm::full::kOutputStateTensor); in lstm_eval()
|