Home
last modified time | relevance | path

Searched refs:kPrevOutputTensor (Results 1 – 4 of 4) sorted by relevance

/frameworks/ml/nn/common/operations/
DQuantizedLSTM.h58 static constexpr int kPrevOutputTensor = 14; variable
DQLSTM.cpp66 constexpr uint32_t kPrevOutputTensor = 18; variable
170 kPrevOutputTensor, in prepare()
297 const Shape outputStateShape = context->getInputShape(kPrevOutputTensor); in prepare()
352 const Shape prevOutputShape = context->getInputShape(kPrevOutputTensor); in prepare()
387 const Shape prevOutputShape = context->getInputShape(kPrevOutputTensor); in execute()
456 reinterpret_cast<const int8_t*>(context->getInputBuffer(kPrevOutputTensor)); in execute()
DQuantizedLSTMTest.cpp77 initializeInputData(inputOperandTypeParams[QuantizedLSTMCell::kPrevOutputTensor], in QuantizedLSTMOpModel()
136 ASSERT_EQ(setInputTensor(&execution, QuantizedLSTMCell::kPrevOutputTensor, prevOutput_), in invoke()
DQuantizedLSTM.cpp242 prevOutput_ = GetInput(operation, operands, kPrevOutputTensor); in QuantizedLSTMCell()
257 auto prevOutput = GetInput(operation, operands, kPrevOutputTensor); in prepare()