Searched refs:kCellToOutputWeightsTensor (Results 1 – 8 of 8) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | UnidirectionalSequenceLSTM.cpp | 56 params.use_peephole = hasTensor(context, kCellToOutputWeightsTensor); in getLSTMParams()
161 if (hasTensor(context, kCellToOutputWeightsTensor)) { in prepare()
162 const Shape cellToOutputShape = context->getInputShape(kCellToOutputWeightsTensor); in prepare()
172 hasTensor(context, kCellToOutputWeightsTensor)) || in prepare()
175 !hasTensor(context, kCellToOutputWeightsTensor)); in prepare()
331 context->getInputBuffer<float>(kCellToOutputWeightsTensor), in execute()
384 context->getInputBuffer<_Float16>(kCellToOutputWeightsTensor), in execute()
|
| D | QLSTM.cpp | 132 if (hasTensor(context, kCellToOutputWeightsTensor)) { in prepare()
133 const Shape cellToOutputShape = context->getInputShape(kCellToOutputWeightsTensor); in prepare()
143 hasTensor(context, kCellToOutputWeightsTensor)) || in prepare()
146 !hasTensor(context, kCellToOutputWeightsTensor)); in prepare()
270 const Shape cellToOutputShape = context->getInputShape(kCellToOutputWeightsTensor); in execute()
330 reinterpret_cast<const int16_t*>(context->getInputBuffer(kCellToOutputWeightsTensor)); in execute()
|
| D | LayerNormLSTMTest.cpp | 240 execution.setInput(LSTMCell::kCellToOutputWeightsTensor, nullptr, 0); in Invoke()
|
| D | LSTMTest.cpp | 227 execution.setInput(LSTMCell::kCellToOutputWeightsTensor, nullptr, 0); in Invoke()
|
| D | LSTM.cpp | 73 GetInput(operation, operands, kCellToOutputWeightsTensor); // optional in LSTMCell()
|
| /packages/modules/NeuralNetworks/common/types/operations/include/ |
| D | UnidirectionalSequenceLSTM.h | 45 constexpr uint32_t kCellToOutputWeightsTensor = 11; // Optional variable
|
| D | QLSTM.h | 44 constexpr uint32_t kCellToOutputWeightsTensor = 11; variable
|
| D | LSTM.h | 74 static constexpr int kCellToOutputWeightsTensor = 11; // Optional variable
|