Searched refs:kInputToCellWeightsTensor (Results 1 – 9 of 9) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | UnidirectionalSequenceLSTM.cpp | 70 kInputToCellWeightsTensor, in prepare()
120 const Shape inputToCellShape = context->getInputShape(kInputToCellWeightsTensor); in prepare()
321 context->getInputBuffer<float>(kInputToCellWeightsTensor), in execute()
374 context->getInputBuffer<_Float16>(kInputToCellWeightsTensor), in execute()
|
| D | QLSTM.cpp | 47 kInputToCellWeightsTensor, in prepare()
91 const Shape inputToCellShape = context->getInputShape(kInputToCellWeightsTensor); in prepare()
264 const Shape inputToCellWeightsShape = context->getInputShape(kInputToCellWeightsTensor); in execute()
317 reinterpret_cast<const int8_t*>(context->getInputBuffer(kInputToCellWeightsTensor)); in execute()
|
| D | QuantizedLSTM.cpp | 229 inputToCellWeights_ = GetInput(operation, operands, kInputToCellWeightsTensor); in QuantizedLSTMCell()
280 auto inputToCellWeights = GetInput(operation, operands, kInputToCellWeightsTensor); in prepare()
|
| D | QuantizedLSTMTest.cpp | 105 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()
|
| /packages/modules/NeuralNetworks/common/types/operations/include/ |
| D | QuantizedLSTM.h | 42 static constexpr int kInputToCellWeightsTensor = 3; variable
|
| D | UnidirectionalSequenceLSTM.h | 33 constexpr uint32_t kInputToCellWeightsTensor = 3; variable
|
| D | QLSTM.h | 31 constexpr uint32_t kInputToCellWeightsTensor = 3; variable
|
| D | LSTM.h | 62 static constexpr int kInputToCellWeightsTensor = 3; variable
|