Searched refs:kRecurrentToInputWeightsTensor (Results 1 – 9 of 9) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | UnidirectionalSequenceLSTM.cpp | 44 constexpr uint32_t kRecurrentToInputWeightsTensor = 5; // Optional variable
231 if (hasTensor(context, kRecurrentToInputWeightsTensor)) { in prepare()
232 const Shape recurrentToInputShape = context->getInputShape(kRecurrentToInputWeightsTensor); in prepare()
250 hasTensor(context, kRecurrentToInputWeightsTensor)) || in prepare()
252 !hasTensor(context, kRecurrentToInputWeightsTensor)); in prepare()
430 context->getInputBuffer<float>(kRecurrentToInputWeightsTensor), in execute()
483 context->getInputBuffer<_Float16>(kRecurrentToInputWeightsTensor), in execute()
|
| D | QLSTM.cpp | 42 constexpr uint32_t kRecurrentToInputWeightsTensor = 5; variable
211 if (hasTensor(context, kRecurrentToInputWeightsTensor)) { in prepare()
212 const Shape recurrentToInputShape = context->getInputShape(kRecurrentToInputWeightsTensor); in prepare()
230 hasTensor(context, kRecurrentToInputWeightsTensor)) || in prepare()
232 !hasTensor(context, kRecurrentToInputWeightsTensor)); in prepare()
370 context->getInputShape(kRecurrentToInputWeightsTensor); in execute()
411 context->getInputBuffer(kRecurrentToInputWeightsTensor)); in execute()
|
| D | QuantizedLSTM.h | 46 static constexpr int kRecurrentToInputWeightsTensor = 5; variable
|
| D | QuantizedLSTM.cpp | 231 recurrentToInputWeights_ = GetInput(operation, operands, kRecurrentToInputWeightsTensor); in QuantizedLSTMCell()
286 auto recurrentToInputWeights = GetInput(operation, operands, kRecurrentToInputWeightsTensor); in prepare()
|
| D | LSTM.h | 67 static constexpr int kRecurrentToInputWeightsTensor = 5; // Optional variable
|
| D | LayerNormLSTMTest.cpp | 229 execution.setInput(LSTMCell::kRecurrentToInputWeightsTensor, nullptr, 0); in Invoke()
|
| D | QuantizedLSTMTest.cpp | 110 ASSERT_EQ(setInputTensor(&execution, QuantizedLSTMCell::kRecurrentToInputWeightsTensor, in invoke()
|
| D | LSTMTest.cpp | 218 execution.setInput(LSTMCell::kRecurrentToInputWeightsTensor, nullptr, 0); in Invoke()
|
| D | LSTM.cpp | 64 GetInput(operation, operands, kRecurrentToInputWeightsTensor); // optional in LSTMCell()
|