Home
last modified time | relevance | path

Searched refs:kInputCellStateTensor (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/
H A Dlstm_utils.h52 kInputCellStateTensor = 19, enumerator
H A Didentify_lstm_split_inputs.cc85 lstm_cell_op->inputs[kInputCellStateTensor] = in Run()
H A Dpropagate_fixed_sizes.cc1099 constexpr int kInputCellStateTensor = 19; in ProcessUnidirectionalSequenceLstmOperator() local
1107 model->GetArray(op->inputs[kInputCellStateTensor]).buffer.reset(); in ProcessUnidirectionalSequenceLstmOperator()
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/tflite/
H A Doperator.cc859 mutating_input_variables[kInputCellStateTensor] = true; in GetMutatingInputVariables()
902 mutating_input_variables[kInputCellStateTensor] = true; in GetMutatingInputVariables()