Home
last modified time | relevance | path

Searched refs:scratch_buffer_ (Results 1 – 2 of 2) sorted by relevance

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DLSTM.h243 RunTimeOperandInfo* scratch_buffer_; variable
H A DLSTM.cpp128 scratch_buffer_ = GetOutput(operation, operands, kScratchBufferTensor); in LSTMCell()
1025 GetBuffer<float>(output_), GetBuffer<float>(scratch_buffer_)); in Eval()
1061 GetBuffer<_Float16>(scratch_buffer_)); in Eval()