Searched refs:kRecurrentToOutputWeightsTensor (Results 1 – 9 of 9) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | UnidirectionalSequenceLSTM.cpp | 74 kRecurrentToOutputWeightsTensor, in prepare()
104 const Shape recurrentToOutputShape = context->getInputShape(kRecurrentToOutputWeightsTensor); in prepare()
327 context->getInputBuffer<float>(kRecurrentToOutputWeightsTensor), in execute()
328 context->getInputShape(kRecurrentToOutputWeightsTensor), in execute()
380 context->getInputBuffer<_Float16>(kRecurrentToOutputWeightsTensor), in execute()
381 context->getInputShape(kRecurrentToOutputWeightsTensor), in execute()
|
| D | QLSTM.cpp | 51 kRecurrentToOutputWeightsTensor, in prepare()
75 const Shape recurrentToOutputShape = context->getInputShape(kRecurrentToOutputWeightsTensor); in prepare()
269 context->getInputShape(kRecurrentToOutputWeightsTensor); in execute()
328 context->getInputBuffer(kRecurrentToOutputWeightsTensor)); in execute()
|
| D | QuantizedLSTM.cpp | 235 recurrentToOutputWeights_ = GetInput(operation, operands, kRecurrentToOutputWeightsTensor); in QuantizedLSTMCell()
290 auto recurrentToOutputWeights = GetInput(operation, operands, kRecurrentToOutputWeightsTensor); in prepare()
|
| D | QuantizedLSTMTest.cpp | 120 ASSERT_EQ(setInputTensor(&execution, QuantizedLSTMCell::kRecurrentToOutputWeightsTensor, in invoke()
|
| D | LSTM.cpp | 67 recurrent_to_output_weights_ = GetInput(operation, operands, kRecurrentToOutputWeightsTensor); in LSTMCell()
316 kRecurrentToOutputWeightsTensor, in Prepare()
|
| /packages/modules/NeuralNetworks/common/types/operations/include/ |
| D | QuantizedLSTM.h | 49 static constexpr int kRecurrentToOutputWeightsTensor = 8; variable
|
| D | UnidirectionalSequenceLSTM.h | 40 constexpr uint32_t kRecurrentToOutputWeightsTensor = 8; variable
|
| D | QLSTM.h | 38 constexpr uint32_t kRecurrentToOutputWeightsTensor = 8; variable
|
| D | LSTM.h | 69 static constexpr int kRecurrentToOutputWeightsTensor = 8; variable
|