/aosp_15_r20/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()
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
H A D | identify_lstm_merge_inputs.cc | 62 int num_cell = model->GetArray(src_op->inputs[kInputToInputWeightsTensor]) in Run() 65 int num_input = model->GetArray(src_op->inputs[kInputToInputWeightsTensor]) in Run() 93 model->GetArray(src_op->inputs[kInputToInputWeightsTensor]), 0, 0); in Run()
|
H A D | lstm_utils.h | 33 kInputToInputWeightsTensor = 1, // Optional enumerator
|
H A D | identify_lstm_split_inputs.cc | 97 model, &(lstm_cell_op->inputs[kInputToInputWeightsTensor]), in Run()
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
H A D | lstm.cc | 183 GetOptionalInputTensor(context, node, kInputToInputWeightsTensor); in PopulateQuantizedLstmParams8x8_16() 471 GetOptionalInputTensor(context, node, kInputToInputWeightsTensor); in PopulateQuantizedLstmParams8x8_8() 877 GetOptionalInputTensor(context, node, kInputToInputWeightsTensor); in CheckInputTensorDimensions() 1166 GetOptionalInputTensor(context, node, kInputToInputWeightsTensor); in PopulatePrecomputedZPTimesWeightsWithBias() 1417 GetOptionalInputTensor(context, node, kInputToInputWeightsTensor); in Prepare() 1602 GetOptionalInputTensor(context, node, kInputToInputWeightsTensor); in Prepare() 1633 GetOptionalInputTensor(context, node, kInputToInputWeightsTensor); in Prepare() 1837 GetOptionalInputTensor(context, node, kInputToInputWeightsTensor); in Eval()
|
H A D | lstm_shared.h | 31 constexpr int kInputToInputWeightsTensor = 1; // Optional variable
|
H A D | unidirectional_sequence_lstm.cc | 97 context, node, lstm::full::kInputToInputWeightsTensor); in PopulateQuantizedLstmParams8x8_16() 430 context, node, lstm::full::kInputToInputWeightsTensor); in CheckInputTensorDimensions() 727 context, node, lstm::full::kInputToInputWeightsTensor); in PopulatePrecomputedZPTimesWeightsWithBias() 970 context, node, lstm::full::kInputToInputWeightsTensor); in Prepare() 1230 context, node, lstm::full::kInputToInputWeightsTensor); in Eval()
|
/aosp_15_r20/packages/modules/NeuralNetworks/common/types/operations/include/ |
D | UnidirectionalSequenceLSTM.h | 31 constexpr uint32_t kInputToInputWeightsTensor = 1; // Optional variable
|
D | QuantizedLSTM.h | 40 static constexpr int kInputToInputWeightsTensor = 1; variable
|
D | QLSTM.h | 29 constexpr uint32_t kInputToInputWeightsTensor = 1; variable
|
D | LSTM.h | 60 static constexpr int kInputToInputWeightsTensor = 1; // Optional variable
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
H A D | lstm_parser.cc | 108 node, tflite::ops::builtin::lstm::full::kInputToInputWeightsTensor); in HasCifg() 509 tflite::ops::builtin::lstm::full::kInputToInputWeightsTensor, in ParseLSTMAttributes()
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
H A D | lstm.cc | 552 context, node, ops::builtin::lstm::full::kInputToInputWeightsTensor); in lstm_eval()
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
H A D | lstm.cc | 474 context, node, ops::builtin::lstm::full::kInputToInputWeightsTensor); in lstm_eval()
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/nnapi/ |
H A D | nnapi_delegate.cc | 5565 constexpr int kInputToInputWeightsTensor = 1; in AddOpsAndTensors() local 5579 if (input_pos == kInputToInputWeightsTensor || in AddOpsAndTensors()
|