Searched refs:kProjectionWeightsTensor (Results 1 – 8 of 8) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | UnidirectionalSequenceLSTM.cpp | 58 params.use_projection_weight = hasTensor(context, kProjectionWeightsTensor); in getLSTMParams()
198 if (hasTensor(context, kProjectionWeightsTensor)) { in prepare()
199 const Shape projectionShape = context->getInputShape(kProjectionWeightsTensor); in prepare()
341 context->getInputBuffer<float>(kProjectionWeightsTensor), in execute()
394 context->getInputBuffer<_Float16>(kProjectionWeightsTensor), in execute()
|
| D | QLSTM.cpp | 169 if (hasTensor(context, kProjectionWeightsTensor)) { in prepare()
170 const Shape projectionShape = context->getInputShape(kProjectionWeightsTensor); in prepare()
272 const Shape projectionWeightsShape = context->getInputShape(kProjectionWeightsTensor); in execute()
337 reinterpret_cast<const int8_t*>(context->getInputBuffer(kProjectionWeightsTensor)); in execute()
|
| D | LayerNormLSTMTest.cpp | 248 execution.setInput(LSTMCell::kProjectionWeightsTensor, nullptr, 0); in Invoke()
|
| D | LSTMTest.cpp | 235 execution.setInput(LSTMCell::kProjectionWeightsTensor, nullptr, 0); in Invoke()
|
| D | LSTM.cpp | 80 projection_weights_ = GetInput(operation, operands, kProjectionWeightsTensor); // optional in LSTMCell()
|
| /packages/modules/NeuralNetworks/common/types/operations/include/ |
| D | UnidirectionalSequenceLSTM.h | 54 constexpr uint32_t kProjectionWeightsTensor = 16; // Optional variable
|
| D | QLSTM.h | 53 constexpr uint32_t kProjectionWeightsTensor = 16; variable
|
| D | LSTM.h | 83 static constexpr int kProjectionWeightsTensor = 16; // Optional variable
|