| /aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | UnidirectionalSequenceLSTM.cpp | 73 kRecurrentToCellWeightsTensor, in prepare()
136 const Shape recurrentToCellShape = context->getInputShape(kRecurrentToCellWeightsTensor); in prepare()
326 context->getInputBuffer<float>(kRecurrentToCellWeightsTensor), in execute()
379 context->getInputBuffer<_Float16>(kRecurrentToCellWeightsTensor), in execute()
|
| D | QLSTM.cpp | 50 kRecurrentToCellWeightsTensor, in prepare()
107 const Shape recurrentToCellShape = context->getInputShape(kRecurrentToCellWeightsTensor); in prepare()
265 const Shape recurrentToCellWeightsShape = context->getInputShape(kRecurrentToCellWeightsTensor); in execute()
319 reinterpret_cast<const int8_t*>(context->getInputBuffer(kRecurrentToCellWeightsTensor)); in execute()
|
| D | QuantizedLSTM.cpp | 234 recurrentToCellWeights_ = GetInput(operation, operands, kRecurrentToCellWeightsTensor); in QuantizedLSTMCell()
289 auto recurrentToCellWeights = GetInput(operation, operands, kRecurrentToCellWeightsTensor); in prepare()
|
| D | QuantizedLSTMTest.cpp | 117 ASSERT_EQ(setInputTensor(&execution, QuantizedLSTMCell::kRecurrentToCellWeightsTensor, in invoke()
|
| D | LSTM.cpp | 66 recurrent_to_cell_weights_ = GetInput(operation, operands, kRecurrentToCellWeightsTensor); in LSTMCell()
315 kRecurrentToCellWeightsTensor, in Prepare()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | lstm_shared.h | 39 constexpr int kRecurrentToCellWeightsTensor = 7; variable
|
| H A D | unidirectional_sequence_lstm.cc | 123 GetInputSafe(context, node, lstm::full::kRecurrentToCellWeightsTensor, in PopulateQuantizedLstmParams8x8_16()
478 GetInputSafe(context, node, lstm::full::kRecurrentToCellWeightsTensor, in CheckInputTensorDimensions()
753 GetInputSafe(context, node, lstm::full::kRecurrentToCellWeightsTensor, in PopulatePrecomputedZPTimesWeightsWithBias()
1256 GetInputSafe(context, node, lstm::full::kRecurrentToCellWeightsTensor, in Eval()
|
| H A D | lstm.cc | 205 GetInputSafe(context, node, kRecurrentToCellWeightsTensor, in PopulateQuantizedLstmParams8x8_16()
493 GetInputSafe(context, node, kRecurrentToCellWeightsTensor, in PopulateQuantizedLstmParams8x8_8()
923 GetInputSafe(context, node, kRecurrentToCellWeightsTensor, in CheckInputTensorDimensions()
1188 GetInputSafe(context, node, kRecurrentToCellWeightsTensor, in PopulatePrecomputedZPTimesWeightsWithBias()
1710 GetInput(context, node, kRecurrentToCellWeightsTensor); in Prepare()
1859 GetInputSafe(context, node, kRecurrentToCellWeightsTensor, in Eval()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| H A D | lstm_utils.h | 39 kRecurrentToCellWeightsTensor = 7, enumerator
|
| H A D | identify_lstm_split_inputs.cc | 114 &(lstm_cell_op->inputs[kRecurrentToCellWeightsTensor]), in Run()
|
| H A D | identify_lstm_merge_inputs.cc | 111 model->GetArray(src_op->inputs[kRecurrentToCellWeightsTensor]), num_cell, in Run()
|
| /aosp_15_r20/packages/modules/NeuralNetworks/common/types/operations/include/ |
| D | UnidirectionalSequenceLSTM.h | 39 constexpr uint32_t kRecurrentToCellWeightsTensor = 7; variable
|
| D | QuantizedLSTM.h | 48 static constexpr int kRecurrentToCellWeightsTensor = 7; variable
|
| D | QLSTM.h | 37 constexpr uint32_t kRecurrentToCellWeightsTensor = 7; variable
|
| D | LSTM.h | 68 static constexpr int kRecurrentToCellWeightsTensor = 7; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
| H A D | lstm.cc | 565 context, node, ops::builtin::lstm::full::kRecurrentToCellWeightsTensor); in lstm_eval()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| H A D | lstm.cc | 505 ops::builtin::lstm::full::kRecurrentToCellWeightsTensor, in lstm_eval()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| H A D | lstm_parser.cc | 522 tflite::ops::builtin::lstm::full::kRecurrentToCellWeightsTensor, in ParseLSTMAttributes()
|