Home
last modified time | relevance | path

Searched refs:kInputToInputWeightsTensor (Results 1 – 9 of 9) sorted by relevance

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DUnidirectionalSequenceLSTM.cpp40 constexpr uint32_t kInputToInputWeightsTensor = 1; // Optional variable
109 params.use_cifg = !hasTensor(context, kInputToInputWeightsTensor); in getLSTMParams()
218 if (hasTensor(context, kInputToInputWeightsTensor)) { in prepare()
219 const Shape inputToInputShape = context->getInputShape(kInputToInputWeightsTensor); in prepare()
252 const bool cifgWeightsAllOrNone = (hasTensor(context, kInputToInputWeightsTensor) && in prepare()
254 (!hasTensor(context, kInputToInputWeightsTensor) && in prepare()
277 const bool cifgUsed = !hasTensor(context, kInputToInputWeightsTensor); in prepare()
403 const bool use_cifg = !hasTensor(context, kInputToInputWeightsTensor); in execute()
428 context->getInputBuffer<float>(kInputToInputWeightsTensor), in execute()
481 context->getInputBuffer<_Float16>(kInputToInputWeightsTensor), in execute()
H A DQLSTM.cpp39 constexpr uint32_t kInputToInputWeightsTensor = 1; variable
196 if (hasTensor(context, kInputToInputWeightsTensor)) { in prepare()
197 const Shape inputToInputShape = context->getInputShape(kInputToInputWeightsTensor); in prepare()
230 const bool cifgWeightsAllOrNone = (hasTensor(context, kInputToInputWeightsTensor) && in prepare()
232 (!hasTensor(context, kInputToInputWeightsTensor) && in prepare()
255 const bool cifgUsed = !hasTensor(context, kInputToInputWeightsTensor); in prepare()
370 const Shape inputToInputWeightsShape = context->getInputShape(kInputToInputWeightsTensor); in execute()
410 reinterpret_cast<const int8_t*>(context->getInputBuffer(kInputToInputWeightsTensor)); in execute()
H A DQuantizedLSTM.h40 static constexpr int kInputToInputWeightsTensor = 1; variable
H A DQuantizedLSTM.cpp223 inputToInputWeights_ = GetInput(operation, operands, kInputToInputWeightsTensor); in QuantizedLSTMCell()
261 auto inputToInputWeights = GetInput(operation, operands, kInputToInputWeightsTensor); in prepare()
H A DLSTM.h61 static constexpr int kInputToInputWeightsTensor = 1; // Optional variable
H A DLayerNormLSTMTest.cpp228 execution.setInput(LSTMCell::kInputToInputWeightsTensor, nullptr, 0); in Invoke()
H A DQuantizedLSTMTest.cpp99 ASSERT_EQ(setInputTensor(&execution, QuantizedLSTMCell::kInputToInputWeightsTensor, in invoke()
H A DLSTMTest.cpp216 execution.setInput(LSTMCell::kInputToInputWeightsTensor, nullptr, 0); in Invoke()
H A DLSTM.cpp58 GetInput(operation, operands, kInputToInputWeightsTensor); // optional in LSTMCell()