Searched refs:kCellToOutputWeightsTensor (Results 1 – 6 of 6) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | UnidirectionalSequenceLSTM.cpp | 52 constexpr uint32_t kCellToOutputWeightsTensor = 11; // Optional variable
109 params.use_peephole = hasTensor(context, kCellToOutputWeightsTensor); in getLSTMParams()
267 if (hasTensor(context, kCellToOutputWeightsTensor)) { in prepare()
268 const Shape cellToOutputShape = context->getInputShape(kCellToOutputWeightsTensor); in prepare()
278 hasTensor(context, kCellToOutputWeightsTensor)) || in prepare()
281 !hasTensor(context, kCellToOutputWeightsTensor)); in prepare()
437 context->getInputBuffer<float>(kCellToOutputWeightsTensor), in execute()
490 context->getInputBuffer<_Float16>(kCellToOutputWeightsTensor), in execute()
|
| D | QLSTM.cpp | 51 constexpr uint32_t kCellToOutputWeightsTensor = 11; variable
247 if (hasTensor(context, kCellToOutputWeightsTensor)) { in prepare()
248 const Shape cellToOutputShape = context->getInputShape(kCellToOutputWeightsTensor); in prepare()
258 hasTensor(context, kCellToOutputWeightsTensor)) || in prepare()
261 !hasTensor(context, kCellToOutputWeightsTensor)); in prepare()
384 const Shape cellToOutputShape = context->getInputShape(kCellToOutputWeightsTensor); in execute()
444 reinterpret_cast<const int16_t*>(context->getInputBuffer(kCellToOutputWeightsTensor)); in execute()
|
| D | LSTM.h | 75 static constexpr int kCellToOutputWeightsTensor = 11; // Optional variable
|
| D | LayerNormLSTMTest.cpp | 239 execution.setInput(LSTMCell::kCellToOutputWeightsTensor, nullptr, 0); in Invoke()
|
| D | LSTMTest.cpp | 228 execution.setInput(LSTMCell::kCellToOutputWeightsTensor, nullptr, 0); in Invoke()
|
| D | LSTM.cpp | 73 GetInput(operation, operands, kCellToOutputWeightsTensor); // optional in LSTMCell()
|