Searched refs:inputToInputWeights_ (Results 1 – 3 of 3) sorted by relevance
/packages/modules/NeuralNetworks/common/cpu_operations/ |
D | QuantizedLSTM.cpp | 227 inputToInputWeights_ = GetInput(operation, operands, kInputToInputWeightsTensor); in QuantizedLSTMCell() 355 const int outputSize = SizeOfDimension(inputToInputWeights_, 0); in concatenateWeights() 357 assignWeightsSubmatrix(inputToInputWeights_, 0 * outputSize, outputSize, weightsDims, weights); in concatenateWeights() 425 convertShapeToDims(activationTempShape), inputToInputWeights_->zeroPoint, in eval()
|
D | QuantizedLSTMTest.cpp | 100 inputToInputWeights_), in invoke() 171 inputToInputWeights_ = inputToInputWeights; in setWeightsAndBiases() 204 std::vector<uint8_t> inputToInputWeights_; member in android::nn::wrapper::QuantizedLSTMOpModel
|
/packages/modules/NeuralNetworks/common/types/operations/include/ |
D | QuantizedLSTM.h | 67 const RunTimeOperandInfo* inputToInputWeights_; variable
|