Searched refs:kRecurrentToCellWeightsTensor (Results 1 – 7 of 7) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | UnidirectionalSequenceLSTM.cpp | 46 constexpr uint32_t kRecurrentToCellWeightsTensor = 7; variable
180 kRecurrentToCellWeightsTensor, in prepare()
242 const Shape recurrentToCellShape = context->getInputShape(kRecurrentToCellWeightsTensor); in prepare()
432 context->getInputBuffer<float>(kRecurrentToCellWeightsTensor), in execute()
485 context->getInputBuffer<_Float16>(kRecurrentToCellWeightsTensor), in execute()
|
| D | QuantizedLSTM.h | 48 static constexpr int kRecurrentToCellWeightsTensor = 7; variable
|
| D | QLSTM.cpp | 44 constexpr uint32_t kRecurrentToCellWeightsTensor = 7; variable
165 kRecurrentToCellWeightsTensor, in prepare()
222 const Shape recurrentToCellShape = context->getInputShape(kRecurrentToCellWeightsTensor); in prepare()
379 const Shape recurrentToCellWeightsShape = context->getInputShape(kRecurrentToCellWeightsTensor); in execute()
433 reinterpret_cast<const int8_t*>(context->getInputBuffer(kRecurrentToCellWeightsTensor)); in execute()
|
| D | QuantizedLSTM.cpp | 233 recurrentToCellWeights_ = GetInput(operation, operands, kRecurrentToCellWeightsTensor); in QuantizedLSTMCell()
288 auto recurrentToCellWeights = GetInput(operation, operands, kRecurrentToCellWeightsTensor); in prepare()
|
| D | LSTM.h | 69 static constexpr int kRecurrentToCellWeightsTensor = 7; variable
|
| D | QuantizedLSTMTest.cpp | 116 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()
|