Home
last modified time | relevance | path

Searched refs:mutating_input_variables (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/tflite/
H A Doperator.cc855 std::vector<bool> mutating_input_variables(op.inputs.size(), false); in GetMutatingInputVariables() local
858 mutating_input_variables[kInputActivationStateTensor] = true; in GetMutatingInputVariables()
859 mutating_input_variables[kInputCellStateTensor] = true; in GetMutatingInputVariables()
863 mutating_input_variables[LstmCellOperator::PREV_ACTIV_INPUT] = true; in GetMutatingInputVariables()
864 mutating_input_variables[LstmCellOperator::PREV_STATE_INPUT] = true; in GetMutatingInputVariables()
868 return mutating_input_variables; in GetMutatingInputVariables()
900 std::vector<bool> mutating_input_variables(op.inputs.size(), false); in GetMutatingInputVariables() local
901 mutating_input_variables[kInputActivationStateTensor] = true; in GetMutatingInputVariables()
902 mutating_input_variables[kInputCellStateTensor] = true; in GetMutatingInputVariables()
903 return mutating_input_variables; in GetMutatingInputVariables()
[all …]
H A Dexport.cc378 std::vector<bool> mutating_input_variables; in ExportOperators() local
393 mutating_input_variables = tflite_op->GetMutatingInputVariables(*op); in ExportOperators()
395 if (!mutating_input_variables.empty()) { in ExportOperators()
397 if (!mutating_input_variables[i]) { in ExportOperators()
416 builder->CreateVector(mutating_input_variables))); in ExportOperators()