Searched refs:kCellToOutputWeightsTensor (Results 1 – 5 of 5) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | UnidirectionalSequenceLSTM.cpp | 50 constexpr uint32_t kCellToOutputWeightsTensor = 11; // Optional variable
102 params.use_peephole = hasTensor(context, kCellToOutputWeightsTensor); in getLSTMParams()
254 if (hasTensor(context, kCellToOutputWeightsTensor)) { in prepare()
255 const Shape cellToOutputShape = context->getInputShape(kCellToOutputWeightsTensor); in prepare()
265 hasTensor(context, kCellToOutputWeightsTensor)) || in prepare()
268 !hasTensor(context, kCellToOutputWeightsTensor)); in prepare()
394 context->getInputBuffer<float>(kCellToOutputWeightsTensor), in execute()
436 context->getInputBuffer<_Float16>(kCellToOutputWeightsTensor), in execute()
|
| D | LSTM.h | 73 static constexpr int kCellToOutputWeightsTensor = 11; // Optional variable
|
| D | LayerNormLSTMTest.cpp | 240 execution.setInput(LSTMCell::kCellToOutputWeightsTensor, nullptr, 0); in Invoke()
|
| D | LSTMTest.cpp | 224 execution.setInput(LSTMCell::kCellToOutputWeightsTensor, nullptr, 0); in Invoke()
|
| D | LSTM.cpp | 68 GetInput(operation, operands, kCellToOutputWeightsTensor); // optional in LSTMCell()
|