Searched refs:kInputToInputWeightsTensor (Results 1 – 9 of 9) sorted by relevance
/aosp12/packages/modules/NeuralNetworks/common/operations/ |
H A D | UnidirectionalSequenceLSTM.cpp | 40 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 D | QLSTM.cpp | 39 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 D | QuantizedLSTM.h | 40 static constexpr int kInputToInputWeightsTensor = 1; variable
|
H A D | QuantizedLSTM.cpp | 223 inputToInputWeights_ = GetInput(operation, operands, kInputToInputWeightsTensor); in QuantizedLSTMCell() 261 auto inputToInputWeights = GetInput(operation, operands, kInputToInputWeightsTensor); in prepare()
|
H A D | LSTM.h | 61 static constexpr int kInputToInputWeightsTensor = 1; // Optional variable
|
H A D | LayerNormLSTMTest.cpp | 228 execution.setInput(LSTMCell::kInputToInputWeightsTensor, nullptr, 0); in Invoke()
|
H A D | QuantizedLSTMTest.cpp | 99 ASSERT_EQ(setInputTensor(&execution, QuantizedLSTMCell::kInputToInputWeightsTensor, in invoke()
|
H A D | LSTMTest.cpp | 216 execution.setInput(LSTMCell::kInputToInputWeightsTensor, nullptr, 0); in Invoke()
|
H A D | LSTM.cpp | 58 GetInput(operation, operands, kInputToInputWeightsTensor); // optional in LSTMCell()
|