Searched defs:forgetGateScratch (Results 1 – 3 of 3) sorted by relevance
13 void LstmImpl(const LstmDescriptor& descriptor, in LstmImpl()
84 …std::unique_ptr<Encoder<float>> forgetGateScratch = MakeEncoder<float>(outputInfo, outputs[0]->Map… in Execute() local
103 …std::unique_ptr<Encoder<float>> forgetGateScratch = MakeEncoder<float>(scratchInfo, forgetGateScra… in Execute() local