| /aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | UnidirectionalSequenceLSTM.cpp | 125 if (hasTensor(context, kRecurrentToInputWeightsTensor)) { in prepare()
126 const Shape recurrentToInputShape = context->getInputShape(kRecurrentToInputWeightsTensor); in prepare()
144 hasTensor(context, kRecurrentToInputWeightsTensor)) || in prepare()
146 !hasTensor(context, kRecurrentToInputWeightsTensor)); in prepare()
324 context->getInputBuffer<float>(kRecurrentToInputWeightsTensor), in execute()
377 context->getInputBuffer<_Float16>(kRecurrentToInputWeightsTensor), in execute()
|
| D | QLSTM.cpp | 96 if (hasTensor(context, kRecurrentToInputWeightsTensor)) { in prepare()
97 const Shape recurrentToInputShape = context->getInputShape(kRecurrentToInputWeightsTensor); in prepare()
115 hasTensor(context, kRecurrentToInputWeightsTensor)) || in prepare()
117 !hasTensor(context, kRecurrentToInputWeightsTensor)); in prepare()
256 context->getInputShape(kRecurrentToInputWeightsTensor); in execute()
297 context->getInputBuffer(kRecurrentToInputWeightsTensor)); in execute()
|
| D | QuantizedLSTM.cpp | 232 recurrentToInputWeights_ = GetInput(operation, operands, kRecurrentToInputWeightsTensor); in QuantizedLSTMCell()
287 auto recurrentToInputWeights = GetInput(operation, operands, kRecurrentToInputWeightsTensor); in prepare()
|
| D | LayerNormLSTMTest.cpp | 230 execution.setInput(LSTMCell::kRecurrentToInputWeightsTensor, nullptr, 0); in Invoke()
|
| D | QuantizedLSTMTest.cpp | 111 ASSERT_EQ(setInputTensor(&execution, QuantizedLSTMCell::kRecurrentToInputWeightsTensor, in invoke()
|
| D | LSTMTest.cpp | 217 execution.setInput(LSTMCell::kRecurrentToInputWeightsTensor, nullptr, 0); in Invoke()
|
| D | LSTM.cpp | 64 GetInput(operation, operands, kRecurrentToInputWeightsTensor); // optional in LSTMCell()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| H A D | identify_lstm_merge_inputs.cc | 69 model->GetArray(src_op->inputs[kRecurrentToInputWeightsTensor]) in Run()
107 model->GetArray(src_op->inputs[kRecurrentToInputWeightsTensor]), 0, in Run()
|
| H A D | lstm_utils.h | 37 kRecurrentToInputWeightsTensor = 5, // Optional enumerator
|
| H A D | identify_lstm_split_inputs.cc | 111 model, &(lstm_cell_op->inputs[kRecurrentToInputWeightsTensor]), in Run()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | lstm_shared.h | 37 constexpr int kRecurrentToInputWeightsTensor = 5; // Optional variable
|
| H A D | unidirectional_sequence_lstm.cc | 114 context, node, lstm::full::kRecurrentToInputWeightsTensor); in PopulateQuantizedLstmParams8x8_16()
455 context, node, lstm::full::kRecurrentToInputWeightsTensor); in CheckInputTensorDimensions()
744 context, node, lstm::full::kRecurrentToInputWeightsTensor); in PopulatePrecomputedZPTimesWeightsWithBias()
1247 context, node, lstm::full::kRecurrentToInputWeightsTensor); in Eval()
|
| H A D | lstm.cc | 198 GetOptionalInputTensor(context, node, kRecurrentToInputWeightsTensor); in PopulateQuantizedLstmParams8x8_16()
486 GetOptionalInputTensor(context, node, kRecurrentToInputWeightsTensor); in PopulateQuantizedLstmParams8x8_8()
898 GetOptionalInputTensor(context, node, kRecurrentToInputWeightsTensor); in CheckInputTensorDimensions()
1181 GetOptionalInputTensor(context, node, kRecurrentToInputWeightsTensor); in PopulatePrecomputedZPTimesWeightsWithBias()
1683 context, node, kRecurrentToInputWeightsTensor); in Prepare()
1852 GetOptionalInputTensor(context, node, kRecurrentToInputWeightsTensor); in Eval()
|
| /aosp_15_r20/packages/modules/NeuralNetworks/common/types/operations/include/ |
| D | UnidirectionalSequenceLSTM.h | 37 constexpr uint32_t kRecurrentToInputWeightsTensor = 5; // Optional variable
|
| D | QuantizedLSTM.h | 46 static constexpr int kRecurrentToInputWeightsTensor = 5; variable
|
| D | QLSTM.h | 35 constexpr uint32_t kRecurrentToInputWeightsTensor = 5; variable
|
| D | LSTM.h | 66 static constexpr int kRecurrentToInputWeightsTensor = 5; // Optional variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
| H A D | lstm.cc | 561 context, node, ops::builtin::lstm::full::kRecurrentToInputWeightsTensor); in lstm_eval()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| H A D | lstm.cc | 494 context, node, ops::builtin::lstm::full::kRecurrentToInputWeightsTensor); in lstm_eval()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| H A D | lstm_parser.cc | 510 tflite::ops::builtin::lstm::full::kRecurrentToInputWeightsTensor, in ParseLSTMAttributes()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| H A D | nnapi_delegate.cc | 5566 constexpr int kRecurrentToInputWeightsTensor = 5; in AddOpsAndTensors() local
5580 input_pos == kRecurrentToInputWeightsTensor || in AddOpsAndTensors()
|