Searched refs:kCellToInputWeightsTensor (Results 1 – 6 of 6) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | UnidirectionalSequenceLSTM.cpp | 50 constexpr uint32_t kCellToInputWeightsTensor = 9; // Optional variable
255 if (hasTensor(context, kCellToInputWeightsTensor)) { in prepare()
256 const Shape cellToInputShape = context->getInputShape(kCellToInputWeightsTensor); in prepare()
276 ((hasTensor(context, kCellToInputWeightsTensor) || cifgUsed) && in prepare()
279 (!hasTensor(context, kCellToInputWeightsTensor) && in prepare()
435 context->getInputBuffer<float>(kCellToInputWeightsTensor), in execute()
488 context->getInputBuffer<_Float16>(kCellToInputWeightsTensor), in execute()
|
| D | QLSTM.cpp | 49 constexpr uint32_t kCellToInputWeightsTensor = 9; variable
235 if (hasTensor(context, kCellToInputWeightsTensor)) { in prepare()
236 const Shape cellToInputShape = context->getInputShape(kCellToInputWeightsTensor); in prepare()
256 ((hasTensor(context, kCellToInputWeightsTensor) || cifgUsed) && in prepare()
259 (!hasTensor(context, kCellToInputWeightsTensor) && in prepare()
371 const Shape cellToInputShape = context->getInputShape(kCellToInputWeightsTensor); in execute()
413 reinterpret_cast<const int16_t*>(context->getInputBuffer(kCellToInputWeightsTensor)); in execute()
|
| D | LayerNormLSTMTest.cpp | 234 execution.setInput(LSTMCell::kCellToInputWeightsTensor, nullptr, 0); in Invoke()
237 execution.setInput(LSTMCell::kCellToInputWeightsTensor, nullptr, 0); in Invoke()
|
| D | LSTMTest.cpp | 223 execution.setInput(LSTMCell::kCellToInputWeightsTensor, nullptr, 0); in Invoke()
226 execution.setInput(LSTMCell::kCellToInputWeightsTensor, nullptr, 0); in Invoke()
|
| D | LSTM.h | 73 static constexpr int kCellToInputWeightsTensor = 9; // Optional variable
|
| D | LSTM.cpp | 69 cell_to_input_weights_ = GetInput(operation, operands, kCellToInputWeightsTensor); // optional in LSTMCell()
|