Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/lite/utils/
Dlstm_utils.h48 class ConvertLSTMCellSimpleToFusedLSTM {
50 explicit ConvertLSTMCellSimpleToFusedLSTM(mlir::FuncOp fused_func_op) in ConvertLSTMCellSimpleToFusedLSTM() function
56 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()
Dlstm_utils.cc184 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 …]
Dlstm_utils_test.cc121 mlir::TFL::ConvertLSTMCellSimpleToFusedLSTM convert(fused_lstm_func_); in TEST_F()
194 mlir::TFL::ConvertLSTMCellSimpleToFusedLSTM convert(fused_lstm_func_cifg_); in TEST_F()
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dprepare_composite_functions_tf.cc259 ConvertLSTMCellSimpleToFusedLSTM convert_lstm_cell_simple(func); in ConvertTFImplements()