Searched refs:kInputToCellWeightsTensor (Results 1 – 7 of 7) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | UnidirectionalSequenceLSTM.cpp | 40 constexpr uint32_t kInputToCellWeightsTensor = 3; variable
177 kInputToCellWeightsTensor, in prepare()
226 const Shape inputToCellShape = context->getInputShape(kInputToCellWeightsTensor); in prepare()
427 context->getInputBuffer<float>(kInputToCellWeightsTensor), in execute()
480 context->getInputBuffer<_Float16>(kInputToCellWeightsTensor), in execute()
|
| D | QuantizedLSTM.h | 42 static constexpr int kInputToCellWeightsTensor = 3; variable
|
| D | QLSTM.cpp | 38 constexpr uint32_t kInputToCellWeightsTensor = 3; variable
162 kInputToCellWeightsTensor, in prepare()
206 const Shape inputToCellShape = context->getInputShape(kInputToCellWeightsTensor); in prepare()
378 const Shape inputToCellWeightsShape = context->getInputShape(kInputToCellWeightsTensor); in execute()
431 reinterpret_cast<const int8_t*>(context->getInputBuffer(kInputToCellWeightsTensor)); in execute()
|
| D | QuantizedLSTM.cpp | 228 inputToCellWeights_ = GetInput(operation, operands, kInputToCellWeightsTensor); in QuantizedLSTMCell()
279 auto inputToCellWeights = GetInput(operation, operands, kInputToCellWeightsTensor); in prepare()
|
| D | LSTM.h | 63 static constexpr int kInputToCellWeightsTensor = 3; variable
|
| D | QuantizedLSTMTest.cpp | 104 ASSERT_EQ(setInputTensor(&execution, QuantizedLSTMCell::kInputToCellWeightsTensor, in invoke()
|
| D | LSTM.cpp | 60 input_to_cell_weights_ = GetInput(operation, operands, kInputToCellWeightsTensor); in LSTMCell()
312 kInputToCellWeightsTensor, in Prepare()
|