Searched refs:ConvertLSTMCellSimpleToFusedLSTM (Results 1 – 4 of 4) sorted by relevance
48 class ConvertLSTMCellSimpleToFusedLSTM {50 explicit ConvertLSTMCellSimpleToFusedLSTM(mlir::FuncOp fused_func_op) in ConvertLSTMCellSimpleToFusedLSTM() function56 ConvertLSTMCellSimpleToFusedLSTM(const ConvertLSTMCellSimpleToFusedLSTM&) =58 ConvertLSTMCellSimpleToFusedLSTM& operator=(59 const ConvertLSTMCellSimpleToFusedLSTM&) = delete;60 virtual ~ConvertLSTMCellSimpleToFusedLSTM() {} in ~ConvertLSTMCellSimpleToFusedLSTM()175 : public ConvertLSTMCellSimpleToFusedLSTM {179 : ConvertLSTMCellSimpleToFusedLSTM(fused_func_op) {} in ConvertLayerNormalizedLSTMCellSimpleToFusedLSTM()
184 void ConvertLSTMCellSimpleToFusedLSTM::SetWeightForInputToCellGate() { in SetWeightForInputToCellGate()192 void ConvertLSTMCellSimpleToFusedLSTM::SetWeightForInputToInputGate() { in SetWeightForInputToInputGate()203 void ConvertLSTMCellSimpleToFusedLSTM::SetWeightForInputToForgetGate() { in SetWeightForInputToForgetGate()212 void ConvertLSTMCellSimpleToFusedLSTM::SetWeightForInputToOutputGate() { in SetWeightForInputToOutputGate()222 void ConvertLSTMCellSimpleToFusedLSTM::SetWeightForRecurrentToCellGate() { in SetWeightForRecurrentToCellGate()230 void ConvertLSTMCellSimpleToFusedLSTM::SetWeightForRecurrentToInputGate() { in SetWeightForRecurrentToInputGate()241 void ConvertLSTMCellSimpleToFusedLSTM::SetWeightForRecurrentToForgetGate() { in SetWeightForRecurrentToForgetGate()250 void ConvertLSTMCellSimpleToFusedLSTM::SetWeightForRecurrentToOutputGate() { in SetWeightForRecurrentToOutputGate()259 void ConvertLSTMCellSimpleToFusedLSTM::SetBiasToCellGate() { in SetBiasToCellGate()266 void ConvertLSTMCellSimpleToFusedLSTM::SetBiasToInputGate() { in SetBiasToInputGate()[all …]
121 mlir::TFL::ConvertLSTMCellSimpleToFusedLSTM convert(fused_lstm_func_); in TEST_F()194 mlir::TFL::ConvertLSTMCellSimpleToFusedLSTM convert(fused_lstm_func_cifg_); in TEST_F()
259 ConvertLSTMCellSimpleToFusedLSTM convert_lstm_cell_simple(func); in ConvertTFImplements()