Home
last modified time | relevance | path

Searched refs:kRecurrentToOutputWeightsTensor (Results 1 – 7 of 7) sorted by relevance

/frameworks/ml/nn/common/operations/
DUnidirectionalSequenceLSTM.cpp47 constexpr uint32_t kRecurrentToOutputWeightsTensor = 8; variable
181 kRecurrentToOutputWeightsTensor, in prepare()
210 const Shape recurrentToOutputShape = context->getInputShape(kRecurrentToOutputWeightsTensor); in prepare()
433 context->getInputBuffer<float>(kRecurrentToOutputWeightsTensor), in execute()
434 context->getInputShape(kRecurrentToOutputWeightsTensor), in execute()
486 context->getInputBuffer<_Float16>(kRecurrentToOutputWeightsTensor), in execute()
487 context->getInputShape(kRecurrentToOutputWeightsTensor), in execute()
DQuantizedLSTM.h49 static constexpr int kRecurrentToOutputWeightsTensor = 8; variable
DQLSTM.cpp45 constexpr uint32_t kRecurrentToOutputWeightsTensor = 8; variable
166 kRecurrentToOutputWeightsTensor, in prepare()
190 const Shape recurrentToOutputShape = context->getInputShape(kRecurrentToOutputWeightsTensor); in prepare()
383 context->getInputShape(kRecurrentToOutputWeightsTensor); in execute()
442 context->getInputBuffer(kRecurrentToOutputWeightsTensor)); in execute()
DQuantizedLSTM.cpp234 recurrentToOutputWeights_ = GetInput(operation, operands, kRecurrentToOutputWeightsTensor); in QuantizedLSTMCell()
289 auto recurrentToOutputWeights = GetInput(operation, operands, kRecurrentToOutputWeightsTensor); in prepare()
DLSTM.h70 static constexpr int kRecurrentToOutputWeightsTensor = 8; variable
DQuantizedLSTMTest.cpp119 ASSERT_EQ(setInputTensor(&execution, QuantizedLSTMCell::kRecurrentToOutputWeightsTensor, in invoke()
DLSTM.cpp67 recurrent_to_output_weights_ = GetInput(operation, operands, kRecurrentToOutputWeightsTensor); in LSTMCell()
316 kRecurrentToOutputWeightsTensor, in Prepare()