Searched refs:kCellToForgetWeightsTensor (Results 1 – 6 of 6) sorted by relevance
| /aosp12/packages/modules/NeuralNetworks/common/operations/ |
| H A D | UnidirectionalSequenceLSTM.cpp | 53 constexpr uint32_t kCellToForgetWeightsTensor = 10; // Optional variable
264 if (hasTensor(context, kCellToForgetWeightsTensor)) { in prepare()
265 const Shape cellToForgetShape = context->getInputShape(kCellToForgetWeightsTensor); in prepare()
280 hasTensor(context, kCellToForgetWeightsTensor) && in prepare()
283 !hasTensor(context, kCellToForgetWeightsTensor) && in prepare()
439 context->getInputBuffer<float>(kCellToForgetWeightsTensor), in execute()
492 context->getInputBuffer<_Float16>(kCellToForgetWeightsTensor), in execute()
|
| H A D | QLSTM.cpp | 53 constexpr uint32_t kCellToForgetWeightsTensor = 10; variable
242 if (hasTensor(context, kCellToForgetWeightsTensor)) { in prepare()
243 const Shape cellToForgetShape = context->getInputShape(kCellToForgetWeightsTensor); in prepare()
258 hasTensor(context, kCellToForgetWeightsTensor) && in prepare()
261 !hasTensor(context, kCellToForgetWeightsTensor) && in prepare()
378 const Shape cellToForgetShape = context->getInputShape(kCellToForgetWeightsTensor); in execute()
426 reinterpret_cast<const int16_t*>(context->getInputBuffer(kCellToForgetWeightsTensor)); in execute()
|
| H A D | LSTM.h | 74 static constexpr int kCellToForgetWeightsTensor = 10; // Optional variable
|
| H A D | LayerNormLSTMTest.cpp | 238 execution.setInput(LSTMCell::kCellToForgetWeightsTensor, nullptr, 0); in Invoke()
|
| H A D | LSTMTest.cpp | 226 execution.setInput(LSTMCell::kCellToForgetWeightsTensor, nullptr, 0); in Invoke()
|
| H A D | LSTM.cpp | 71 GetInput(operation, operands, kCellToForgetWeightsTensor); // optional in LSTMCell()
|