Searched refs:kCellToOutputWeightsTensor (Results 1 – 6 of 6) sorted by relevance
| /aosp12/packages/modules/NeuralNetworks/common/operations/ |
| H A D | UnidirectionalSequenceLSTM.cpp | 54 constexpr uint32_t kCellToOutputWeightsTensor = 11; // Optional variable
110 params.use_peephole = hasTensor(context, kCellToOutputWeightsTensor); in getLSTMParams()
270 if (hasTensor(context, kCellToOutputWeightsTensor)) { in prepare()
271 const Shape cellToOutputShape = context->getInputShape(kCellToOutputWeightsTensor); in prepare()
281 hasTensor(context, kCellToOutputWeightsTensor)) || in prepare()
284 !hasTensor(context, kCellToOutputWeightsTensor)); in prepare()
440 context->getInputBuffer<float>(kCellToOutputWeightsTensor), in execute()
493 context->getInputBuffer<_Float16>(kCellToOutputWeightsTensor), in execute()
|
| H A D | QLSTM.cpp | 54 constexpr uint32_t kCellToOutputWeightsTensor = 11; variable
248 if (hasTensor(context, kCellToOutputWeightsTensor)) { in prepare()
249 const Shape cellToOutputShape = context->getInputShape(kCellToOutputWeightsTensor); in prepare()
259 hasTensor(context, kCellToOutputWeightsTensor)) || in prepare()
262 !hasTensor(context, kCellToOutputWeightsTensor)); in prepare()
386 const Shape cellToOutputShape = context->getInputShape(kCellToOutputWeightsTensor); in execute()
446 reinterpret_cast<const int16_t*>(context->getInputBuffer(kCellToOutputWeightsTensor)); in execute()
|
| H A D | LSTM.h | 75 static constexpr int kCellToOutputWeightsTensor = 11; // Optional variable
|
| H A D | LayerNormLSTMTest.cpp | 239 execution.setInput(LSTMCell::kCellToOutputWeightsTensor, nullptr, 0); in Invoke()
|
| H A D | LSTMTest.cpp | 227 execution.setInput(LSTMCell::kCellToOutputWeightsTensor, nullptr, 0); in Invoke()
|
| H A D | LSTM.cpp | 73 GetInput(operation, operands, kCellToOutputWeightsTensor); // optional in LSTMCell()
|