Home
last modified time | relevance | path

Searched refs:kCellToInputWeightsTensor (Results 1 – 6 of 6) sorted by relevance

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DUnidirectionalSequenceLSTM.cpp52 constexpr uint32_t kCellToInputWeightsTensor = 9; // Optional variable
258 if (hasTensor(context, kCellToInputWeightsTensor)) { in prepare()
259 const Shape cellToInputShape = context->getInputShape(kCellToInputWeightsTensor); in prepare()
279 ((hasTensor(context, kCellToInputWeightsTensor) || cifgUsed) && in prepare()
282 (!hasTensor(context, kCellToInputWeightsTensor) && in prepare()
438 context->getInputBuffer<float>(kCellToInputWeightsTensor), in execute()
491 context->getInputBuffer<_Float16>(kCellToInputWeightsTensor), in execute()
H A DQLSTM.cpp52 constexpr uint32_t kCellToInputWeightsTensor = 9; variable
236 if (hasTensor(context, kCellToInputWeightsTensor)) { in prepare()
237 const Shape cellToInputShape = context->getInputShape(kCellToInputWeightsTensor); in prepare()
257 ((hasTensor(context, kCellToInputWeightsTensor) || cifgUsed) && in prepare()
260 (!hasTensor(context, kCellToInputWeightsTensor) && in prepare()
373 const Shape cellToInputShape = context->getInputShape(kCellToInputWeightsTensor); in execute()
415 reinterpret_cast<const int16_t*>(context->getInputBuffer(kCellToInputWeightsTensor)); in execute()
H A DLayerNormLSTMTest.cpp234 execution.setInput(LSTMCell::kCellToInputWeightsTensor, nullptr, 0); in Invoke()
237 execution.setInput(LSTMCell::kCellToInputWeightsTensor, nullptr, 0); in Invoke()
H A DLSTMTest.cpp222 execution.setInput(LSTMCell::kCellToInputWeightsTensor, nullptr, 0); in Invoke()
225 execution.setInput(LSTMCell::kCellToInputWeightsTensor, nullptr, 0); in Invoke()
H A DLSTM.h73 static constexpr int kCellToInputWeightsTensor = 9; // Optional variable
H A DLSTM.cpp69 cell_to_input_weights_ = GetInput(operation, operands, kCellToInputWeightsTensor); // optional in LSTMCell()