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.cpp128 scratch_buffer_ = GetOutput(operation, operands, kScratchBufferTensor); in LSTMCell()
1029 GetBuffer<float>(output_), GetBuffer<float>(scratch_buffer_)); in Eval()
1065 GetBuffer<_Float16>(scratch_buffer_)); in Eval()