Searched refs:kRecurrentToInputWeightsTensor (Results 1 – 11 of 11) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | UnidirectionalSequenceLSTM.cpp | 125 if (hasTensor(context, kRecurrentToInputWeightsTensor)) { in prepare()
126 const Shape recurrentToInputShape = context->getInputShape(kRecurrentToInputWeightsTensor); in prepare()
144 hasTensor(context, kRecurrentToInputWeightsTensor)) || in prepare()
146 !hasTensor(context, kRecurrentToInputWeightsTensor)); in prepare()
324 context->getInputBuffer<float>(kRecurrentToInputWeightsTensor), in execute()
377 context->getInputBuffer<_Float16>(kRecurrentToInputWeightsTensor), in execute()
|
| D | QLSTM.cpp | 96 if (hasTensor(context, kRecurrentToInputWeightsTensor)) { in prepare()
97 const Shape recurrentToInputShape = context->getInputShape(kRecurrentToInputWeightsTensor); in prepare()
115 hasTensor(context, kRecurrentToInputWeightsTensor)) || in prepare()
117 !hasTensor(context, kRecurrentToInputWeightsTensor)); in prepare()
256 context->getInputShape(kRecurrentToInputWeightsTensor); in execute()
297 context->getInputBuffer(kRecurrentToInputWeightsTensor)); in execute()
|
| D | QuantizedLSTM.cpp | 232 recurrentToInputWeights_ = GetInput(operation, operands, kRecurrentToInputWeightsTensor); in QuantizedLSTMCell()
287 auto recurrentToInputWeights = GetInput(operation, operands, kRecurrentToInputWeightsTensor); in prepare()
|
| D | LayerNormLSTMTest.cpp | 230 execution.setInput(LSTMCell::kRecurrentToInputWeightsTensor, nullptr, 0); in Invoke()
|
| D | QuantizedLSTMTest.cpp | 111 ASSERT_EQ(setInputTensor(&execution, QuantizedLSTMCell::kRecurrentToInputWeightsTensor, in invoke()
|
| D | LSTMTest.cpp | 217 execution.setInput(LSTMCell::kRecurrentToInputWeightsTensor, nullptr, 0); in Invoke()
|
| D | LSTM.cpp | 64 GetInput(operation, operands, kRecurrentToInputWeightsTensor); // optional in LSTMCell()
|
| /packages/modules/NeuralNetworks/common/types/operations/include/ |
| D | QuantizedLSTM.h | 46 static constexpr int kRecurrentToInputWeightsTensor = 5; variable
|
| D | UnidirectionalSequenceLSTM.h | 37 constexpr uint32_t kRecurrentToInputWeightsTensor = 5; // Optional variable
|
| D | QLSTM.h | 35 constexpr uint32_t kRecurrentToInputWeightsTensor = 5; variable
|
| D | LSTM.h | 66 static constexpr int kRecurrentToInputWeightsTensor = 5; // Optional variable
|