Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DQuantizedLSTM.cpp226 inputToInputWeights_ = GetInput(operation, operands, kInputToInputWeightsTensor); in QuantizedLSTMCell()
354 const int outputSize = SizeOfDimension(inputToInputWeights_, 0); in concatenateWeights()
356 assignWeightsSubmatrix(inputToInputWeights_, 0 * outputSize, outputSize, weightsDims, weights); in concatenateWeights()
424 convertShapeToDims(activationTempShape), inputToInputWeights_->zeroPoint, in eval()
DQuantizedLSTM.h67 const RunTimeOperandInfo* inputToInputWeights_; variable
DQuantizedLSTMTest.cpp99 inputToInputWeights_), in invoke()
170 inputToInputWeights_ = inputToInputWeights; in setWeightsAndBiases()
203 std::vector<uint8_t> inputToInputWeights_; member in android::nn::wrapper::QuantizedLSTMOpModel