Searched refs:kCellToForgetWeightsTensor (Results 1 – 6 of 6) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | UnidirectionalSequenceLSTM.cpp | 51 constexpr uint32_t kCellToForgetWeightsTensor = 10; // Optional variable
261 if (hasTensor(context, kCellToForgetWeightsTensor)) { in prepare()
262 const Shape cellToForgetShape = context->getInputShape(kCellToForgetWeightsTensor); in prepare()
277 hasTensor(context, kCellToForgetWeightsTensor) && in prepare()
280 !hasTensor(context, kCellToForgetWeightsTensor) && in prepare()
436 context->getInputBuffer<float>(kCellToForgetWeightsTensor), in execute()
489 context->getInputBuffer<_Float16>(kCellToForgetWeightsTensor), in execute()
|
| D | QLSTM.cpp | 50 constexpr uint32_t kCellToForgetWeightsTensor = 10; variable
241 if (hasTensor(context, kCellToForgetWeightsTensor)) { in prepare()
242 const Shape cellToForgetShape = context->getInputShape(kCellToForgetWeightsTensor); in prepare()
257 hasTensor(context, kCellToForgetWeightsTensor) && in prepare()
260 !hasTensor(context, kCellToForgetWeightsTensor) && in prepare()
376 const Shape cellToForgetShape = context->getInputShape(kCellToForgetWeightsTensor); in execute()
424 reinterpret_cast<const int16_t*>(context->getInputBuffer(kCellToForgetWeightsTensor)); in execute()
|
| D | LSTM.h | 74 static constexpr int kCellToForgetWeightsTensor = 10; // Optional variable
|
| D | LayerNormLSTMTest.cpp | 238 execution.setInput(LSTMCell::kCellToForgetWeightsTensor, nullptr, 0); in Invoke()
|
| D | LSTMTest.cpp | 227 execution.setInput(LSTMCell::kCellToForgetWeightsTensor, nullptr, 0); in Invoke()
|
| D | LSTM.cpp | 71 GetInput(operation, operands, kCellToForgetWeightsTensor); // optional in LSTMCell()
|