Searched refs:kRecurrentToCellWeightsTensor (Results 1 – 9 of 9) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | UnidirectionalSequenceLSTM.cpp | 73 kRecurrentToCellWeightsTensor, in prepare()
136 const Shape recurrentToCellShape = context->getInputShape(kRecurrentToCellWeightsTensor); in prepare()
326 context->getInputBuffer<float>(kRecurrentToCellWeightsTensor), in execute()
379 context->getInputBuffer<_Float16>(kRecurrentToCellWeightsTensor), in execute()
|
| D | QLSTM.cpp | 50 kRecurrentToCellWeightsTensor, in prepare()
107 const Shape recurrentToCellShape = context->getInputShape(kRecurrentToCellWeightsTensor); in prepare()
265 const Shape recurrentToCellWeightsShape = context->getInputShape(kRecurrentToCellWeightsTensor); in execute()
319 reinterpret_cast<const int8_t*>(context->getInputBuffer(kRecurrentToCellWeightsTensor)); in execute()
|
| D | QuantizedLSTM.cpp | 234 recurrentToCellWeights_ = GetInput(operation, operands, kRecurrentToCellWeightsTensor); in QuantizedLSTMCell()
289 auto recurrentToCellWeights = GetInput(operation, operands, kRecurrentToCellWeightsTensor); in prepare()
|
| D | QuantizedLSTMTest.cpp | 117 ASSERT_EQ(setInputTensor(&execution, QuantizedLSTMCell::kRecurrentToCellWeightsTensor, in invoke()
|
| D | LSTM.cpp | 66 recurrent_to_cell_weights_ = GetInput(operation, operands, kRecurrentToCellWeightsTensor); in LSTMCell()
315 kRecurrentToCellWeightsTensor, in Prepare()
|
| /packages/modules/NeuralNetworks/common/types/operations/include/ |
| D | QuantizedLSTM.h | 48 static constexpr int kRecurrentToCellWeightsTensor = 7; variable
|
| D | UnidirectionalSequenceLSTM.h | 39 constexpr uint32_t kRecurrentToCellWeightsTensor = 7; variable
|
| D | QLSTM.h | 37 constexpr uint32_t kRecurrentToCellWeightsTensor = 7; variable
|
| D | LSTM.h | 68 static constexpr int kRecurrentToCellWeightsTensor = 7; variable
|