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