Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DLSTM.h243 RunTimeOperandInfo* scratch_buffer_; variable
DLSTM.cpp119 scratch_buffer_ = GetOutput(operation, operands, kScratchBufferTensor); in LSTMCell()
987 GetBuffer<float>(output_), GetBuffer<float>(scratch_buffer_)); in Eval()
1023 GetBuffer<_Float16>(scratch_buffer_)); in Eval()