Searched refs:kInputToInputWeightsTensor (Results 1 – 11 of 11) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | UnidirectionalSequenceLSTM.cpp | 55 params.use_cifg = !hasTensor(context, kInputToInputWeightsTensor); in getLSTMParams()
109 if (hasTensor(context, kInputToInputWeightsTensor)) { in prepare()
110 const Shape inputToInputShape = context->getInputShape(kInputToInputWeightsTensor); in prepare()
143 const bool cifgWeightsAllOrNone = (hasTensor(context, kInputToInputWeightsTensor) && in prepare()
145 (!hasTensor(context, kInputToInputWeightsTensor) && in prepare()
168 const bool cifgUsed = !hasTensor(context, kInputToInputWeightsTensor); in prepare()
294 const bool use_cifg = !hasTensor(context, kInputToInputWeightsTensor); in execute()
319 context->getInputBuffer<float>(kInputToInputWeightsTensor), in execute()
372 context->getInputBuffer<_Float16>(kInputToInputWeightsTensor), in execute()
|
| D | QLSTM.cpp | 80 if (hasTensor(context, kInputToInputWeightsTensor)) { in prepare()
81 const Shape inputToInputShape = context->getInputShape(kInputToInputWeightsTensor); in prepare()
114 const bool cifgWeightsAllOrNone = (hasTensor(context, kInputToInputWeightsTensor) && in prepare()
116 (!hasTensor(context, kInputToInputWeightsTensor) && in prepare()
139 const bool cifgUsed = !hasTensor(context, kInputToInputWeightsTensor); in prepare()
254 const Shape inputToInputWeightsShape = context->getInputShape(kInputToInputWeightsTensor); in execute()
294 reinterpret_cast<const int8_t*>(context->getInputBuffer(kInputToInputWeightsTensor)); in execute()
|
| D | QuantizedLSTM.cpp | 227 inputToInputWeights_ = GetInput(operation, operands, kInputToInputWeightsTensor); in QuantizedLSTMCell()
265 auto inputToInputWeights = GetInput(operation, operands, kInputToInputWeightsTensor); in prepare()
|
| D | LayerNormLSTMTest.cpp | 229 execution.setInput(LSTMCell::kInputToInputWeightsTensor, nullptr, 0); in Invoke()
|
| D | QuantizedLSTMTest.cpp | 99 ASSERT_EQ(setInputTensor(&execution, QuantizedLSTMCell::kInputToInputWeightsTensor, in invoke()
|
| D | LSTMTest.cpp | 216 execution.setInput(LSTMCell::kInputToInputWeightsTensor, nullptr, 0); in Invoke()
|
| D | LSTM.cpp | 58 GetInput(operation, operands, kInputToInputWeightsTensor); // optional in LSTMCell()
|
| /packages/modules/NeuralNetworks/common/types/operations/include/ |
| D | QuantizedLSTM.h | 40 static constexpr int kInputToInputWeightsTensor = 1; variable
|
| D | UnidirectionalSequenceLSTM.h | 31 constexpr uint32_t kInputToInputWeightsTensor = 1; // Optional variable
|
| D | QLSTM.h | 29 constexpr uint32_t kInputToInputWeightsTensor = 1; variable
|
| D | LSTM.h | 60 static constexpr int kInputToInputWeightsTensor = 1; // Optional variable
|