Searched refs:QuantizedLSTMCell (Results 1 – 4 of 4) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | QuantizedLSTMTest.cpp | 63 inputOperandTypeParams[QuantizedLSTMCell::kPrevCellStateTensor].shape[1]; in QuantizedLSTMOpModel()
77 initializeInputData(inputOperandTypeParams[QuantizedLSTMCell::kInputTensor], &input_); in QuantizedLSTMOpModel()
78 initializeInputData(inputOperandTypeParams[QuantizedLSTMCell::kPrevOutputTensor], in QuantizedLSTMOpModel()
80 initializeInputData(inputOperandTypeParams[QuantizedLSTMCell::kPrevCellStateTensor], in QuantizedLSTMOpModel()
97 ASSERT_EQ(setInputTensor(&execution, QuantizedLSTMCell::kInputTensor, input_), in invoke()
99 ASSERT_EQ(setInputTensor(&execution, QuantizedLSTMCell::kInputToInputWeightsTensor, in invoke()
102 ASSERT_EQ(setInputTensor(&execution, QuantizedLSTMCell::kInputToForgetWeightsTensor, in invoke()
105 ASSERT_EQ(setInputTensor(&execution, QuantizedLSTMCell::kInputToCellWeightsTensor, in invoke()
108 ASSERT_EQ(setInputTensor(&execution, QuantizedLSTMCell::kInputToOutputWeightsTensor, in invoke()
111 ASSERT_EQ(setInputTensor(&execution, QuantizedLSTMCell::kRecurrentToInputWeightsTensor, in invoke()
[all …]
|
| D | QuantizedLSTM.cpp | 224 QuantizedLSTMCell::QuantizedLSTMCell(const Operation& operation, RunTimeOperandInfo* operands) { in QuantizedLSTMCell() function in android::nn::QuantizedLSTMCell
249 bool QuantizedLSTMCell::prepare(const Operation& operation, RunTimeOperandInfo* operands, in prepare()
353 void QuantizedLSTMCell::concatenateWeights(const std::vector<uint32_t>& weightsDims, in concatenateWeights()
369 void QuantizedLSTMCell::concatenateBiases(uint32_t outputSize, int32_t* bias) { in concatenateBiases()
378 bool QuantizedLSTMCell::eval() { in eval()
|
| /packages/modules/NeuralNetworks/common/types/operations/include/ |
| D | QuantizedLSTM.h | 29 class QuantizedLSTMCell {
31 QuantizedLSTMCell(const Operation& operation, RunTimeOperandInfo* operands);
|
| /packages/modules/NeuralNetworks/common/ |
| D | CpuExecutor.cpp | 1621 operands[outs[QuantizedLSTMCell::kCellStateOutTensor]]; in executeOperation()
1622 RunTimeOperandInfo& output = operands[outs[QuantizedLSTMCell::kOutputTensor]]; in executeOperation()
1625 QuantizedLSTMCell quantizedLSTMCell(operation, operands); in executeOperation()
1627 success = QuantizedLSTMCell::prepare(operation, operands, &cellStateOutShape, in executeOperation()
|