Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/utils/
H A Dlstm_utils.h80 void SetWeightForInputToForgetGate();
H A Dlstm_utils.cc206 void ConvertLSTMCellSimpleToFusedLSTM::SetWeightForInputToForgetGate() { in SetWeightForInputToForgetGate() function in mlir::TFL::ConvertLSTMCellSimpleToFusedLSTM
353 SetWeightForInputToForgetGate(); in GenerateFusedOpOperands()