Searched defs:forget_gate (Results 1 – 5 of 5) sorted by relevance
251 Value* forget_gate, Value* input_gate, in BuildCellStateUpdate()465 Value* forget_gate; in ParseLSTMAttributes() local
158 auto forget_gate = CreateGate( in LoadForLSTMOp() local
101 const float* input_gate, float* forget_gate, in UpdateLstmCellFloat()
219 const float* input_gate, float* forget_gate, in UpdateLstmCellFloat()573 int16_t* forget_gate, const int16_t* cell_gate, in UpdateLstmCellInteger()
535 const float forget_gate = in LstmCell() local