Home
last modified time | relevance | path

Searched refs:kOutputStateTensor (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dlstm_shared.h60 constexpr int kOutputStateTensor = 18; variable
H A Dunidirectional_sequence_lstm.cc151 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 Dlstm.cc232 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 Dlstm_utils.h58 kOutputStateTensor = 0, enumerator
H A Didentify_lstm_split_inputs.cc157 lstm_cell_op->outputs[kOutputStateTensor] = in Run()
H A Didentify_lstm_merge_inputs.cc164 src_op->outputs[kOutputStateTensor]; in Run()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/
H A Dlstm_parser.cc481 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 Dlstm.cc607 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 Dlstm.cc563 context, node, ops::builtin::lstm::full::kOutputStateTensor); in lstm_eval()