| /aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | UnidirectionalSequenceLSTM.cpp | 74 kRecurrentToOutputWeightsTensor, in prepare()
104 const Shape recurrentToOutputShape = context->getInputShape(kRecurrentToOutputWeightsTensor); in prepare()
327 context->getInputBuffer<float>(kRecurrentToOutputWeightsTensor), in execute()
328 context->getInputShape(kRecurrentToOutputWeightsTensor), in execute()
380 context->getInputBuffer<_Float16>(kRecurrentToOutputWeightsTensor), in execute()
381 context->getInputShape(kRecurrentToOutputWeightsTensor), in execute()
|
| D | QLSTM.cpp | 51 kRecurrentToOutputWeightsTensor, in prepare()
75 const Shape recurrentToOutputShape = context->getInputShape(kRecurrentToOutputWeightsTensor); in prepare()
269 context->getInputShape(kRecurrentToOutputWeightsTensor); in execute()
328 context->getInputBuffer(kRecurrentToOutputWeightsTensor)); in execute()
|
| D | QuantizedLSTM.cpp | 235 recurrentToOutputWeights_ = GetInput(operation, operands, kRecurrentToOutputWeightsTensor); in QuantizedLSTMCell()
290 auto recurrentToOutputWeights = GetInput(operation, operands, kRecurrentToOutputWeightsTensor); in prepare()
|
| D | QuantizedLSTMTest.cpp | 120 ASSERT_EQ(setInputTensor(&execution, QuantizedLSTMCell::kRecurrentToOutputWeightsTensor, in invoke()
|
| D | LSTM.cpp | 67 recurrent_to_output_weights_ = GetInput(operation, operands, kRecurrentToOutputWeightsTensor); in LSTMCell()
316 kRecurrentToOutputWeightsTensor, in Prepare()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | lstm_shared.h | 40 constexpr int kRecurrentToOutputWeightsTensor = 8; variable
|
| H A D | unidirectional_sequence_lstm.cc | 128 GetInputSafe(context, node, lstm::full::kRecurrentToOutputWeightsTensor, in PopulateQuantizedLstmParams8x8_16()
758 GetInputSafe(context, node, lstm::full::kRecurrentToOutputWeightsTensor, in PopulatePrecomputedZPTimesWeightsWithBias()
910 GetInputSafe(context, node, lstm::full::kRecurrentToOutputWeightsTensor, in Prepare()
1261 GetInputSafe(context, node, lstm::full::kRecurrentToOutputWeightsTensor, in Eval()
|
| H A D | lstm.cc | 209 GetInputSafe(context, node, kRecurrentToOutputWeightsTensor, in PopulateQuantizedLstmParams8x8_16()
497 GetInputSafe(context, node, kRecurrentToOutputWeightsTensor, in PopulateQuantizedLstmParams8x8_8()
1192 GetInputSafe(context, node, kRecurrentToOutputWeightsTensor, in PopulatePrecomputedZPTimesWeightsWithBias()
1335 GetInputSafe(context, node, kRecurrentToOutputWeightsTensor, in Prepare()
1723 GetInput(context, node, kRecurrentToOutputWeightsTensor); in Prepare()
1863 GetInputSafe(context, node, kRecurrentToOutputWeightsTensor, in Eval()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| H A D | lstm_utils.h | 40 kRecurrentToOutputWeightsTensor = 8, enumerator
|
| H A D | identify_lstm_split_inputs.cc | 122 &(lstm_cell_op->inputs[kRecurrentToOutputWeightsTensor]), in Run()
|
| H A D | identify_lstm_merge_inputs.cc | 119 model->GetArray(src_op->inputs[kRecurrentToOutputWeightsTensor]), in Run()
|
| /aosp_15_r20/packages/modules/NeuralNetworks/common/types/operations/include/ |
| D | UnidirectionalSequenceLSTM.h | 40 constexpr uint32_t kRecurrentToOutputWeightsTensor = 8; variable
|
| D | QuantizedLSTM.h | 49 static constexpr int kRecurrentToOutputWeightsTensor = 8; variable
|
| D | QLSTM.h | 38 constexpr uint32_t kRecurrentToOutputWeightsTensor = 8; variable
|
| D | LSTM.h | 69 static constexpr int kRecurrentToOutputWeightsTensor = 8; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
| H A D | lstm.cc | 567 context, node, ops::builtin::lstm::full::kRecurrentToOutputWeightsTensor); in lstm_eval()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| H A D | lstm.cc | 511 ops::builtin::lstm::full::kRecurrentToOutputWeightsTensor, in lstm_eval()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| H A D | lstm_parser.cc | 538 tflite::ops::builtin::lstm::full::kRecurrentToOutputWeightsTensor, in ParseLSTMAttributes()
|