Home
last modified time | relevance | path

Searched refs:inputToInputWeights_ (Results 1 – 3 of 3) sorted by relevance

/packages/modules/NeuralNetworks/common/cpu_operations/
DQuantizedLSTM.cpp227 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()
DQuantizedLSTMTest.cpp100 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/
DQuantizedLSTM.h67 const RunTimeOperandInfo* inputToInputWeights_; variable