Searched refs:ConvertLSTMCellSimpleToFusedLSTM (Results 1 – 4 of 4) sorted by relevance
187 void ConvertLSTMCellSimpleToFusedLSTM::SetWeightForInputToCellGate() { in SetWeightForInputToCellGate()195 void ConvertLSTMCellSimpleToFusedLSTM::SetWeightForInputToInputGate() { in SetWeightForInputToInputGate()206 void ConvertLSTMCellSimpleToFusedLSTM::SetWeightForInputToForgetGate() { in SetWeightForInputToForgetGate()215 void ConvertLSTMCellSimpleToFusedLSTM::SetWeightForInputToOutputGate() { in SetWeightForInputToOutputGate()225 void ConvertLSTMCellSimpleToFusedLSTM::SetWeightForRecurrentToCellGate() { in SetWeightForRecurrentToCellGate()233 void ConvertLSTMCellSimpleToFusedLSTM::SetWeightForRecurrentToInputGate() { in SetWeightForRecurrentToInputGate()244 void ConvertLSTMCellSimpleToFusedLSTM::SetWeightForRecurrentToForgetGate() { in SetWeightForRecurrentToForgetGate()253 void ConvertLSTMCellSimpleToFusedLSTM::SetWeightForRecurrentToOutputGate() { in SetWeightForRecurrentToOutputGate()262 void ConvertLSTMCellSimpleToFusedLSTM::SetBiasToCellGate() { in SetBiasToCellGate()269 void ConvertLSTMCellSimpleToFusedLSTM::SetBiasToInputGate() { in SetBiasToInputGate()[all …]
49 class ConvertLSTMCellSimpleToFusedLSTM {51 explicit ConvertLSTMCellSimpleToFusedLSTM(mlir::func::FuncOp fused_func_op) in ConvertLSTMCellSimpleToFusedLSTM() function57 ConvertLSTMCellSimpleToFusedLSTM(const ConvertLSTMCellSimpleToFusedLSTM&) =59 ConvertLSTMCellSimpleToFusedLSTM& operator=(60 const ConvertLSTMCellSimpleToFusedLSTM&) = delete;61 virtual ~ConvertLSTMCellSimpleToFusedLSTM() {} in ~ConvertLSTMCellSimpleToFusedLSTM()176 : public ConvertLSTMCellSimpleToFusedLSTM {180 : ConvertLSTMCellSimpleToFusedLSTM(fused_func_op) {} in ConvertLayerNormalizedLSTMCellSimpleToFusedLSTM()
121 mlir::TFL::ConvertLSTMCellSimpleToFusedLSTM convert(fused_lstm_func_); in TEST_F()195 mlir::TFL::ConvertLSTMCellSimpleToFusedLSTM convert(fused_lstm_func_cifg_); in TEST_F()
308 ConvertLSTMCellSimpleToFusedLSTM convert_lstm_cell_simple(func); in ConvertTFImplements()