Home
last modified time | relevance | path

Searched refs:kInputToInputWeightsTensor (Results 1 – 21 of 21) sorted by relevance

/aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/
DUnidirectionalSequenceLSTM.cpp55 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()
DQLSTM.cpp80 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()
DQuantizedLSTM.cpp227 inputToInputWeights_ = GetInput(operation, operands, kInputToInputWeightsTensor); in QuantizedLSTMCell()
265 auto inputToInputWeights = GetInput(operation, operands, kInputToInputWeightsTensor); in prepare()
DLayerNormLSTMTest.cpp229 execution.setInput(LSTMCell::kInputToInputWeightsTensor, nullptr, 0); in Invoke()
DQuantizedLSTMTest.cpp99 ASSERT_EQ(setInputTensor(&execution, QuantizedLSTMCell::kInputToInputWeightsTensor, in invoke()
DLSTMTest.cpp216 execution.setInput(LSTMCell::kInputToInputWeightsTensor, nullptr, 0); in Invoke()
DLSTM.cpp58 GetInput(operation, operands, kInputToInputWeightsTensor); // optional in LSTMCell()
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/
H A Didentify_lstm_merge_inputs.cc62 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 Dlstm_utils.h33 kInputToInputWeightsTensor = 1, // Optional enumerator
H A Didentify_lstm_split_inputs.cc97 model, &(lstm_cell_op->inputs[kInputToInputWeightsTensor]), in Run()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dlstm.cc183 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 Dlstm_shared.h31 constexpr int kInputToInputWeightsTensor = 1; // Optional variable
H A Dunidirectional_sequence_lstm.cc97 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/
DUnidirectionalSequenceLSTM.h31 constexpr uint32_t kInputToInputWeightsTensor = 1; // Optional variable
DQuantizedLSTM.h40 static constexpr int kInputToInputWeightsTensor = 1; variable
DQLSTM.h29 constexpr uint32_t kInputToInputWeightsTensor = 1; variable
DLSTM.h60 static constexpr int kInputToInputWeightsTensor = 1; // Optional variable
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/
H A Dlstm_parser.cc108 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 Dlstm.cc552 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 Dlstm.cc474 context, node, ops::builtin::lstm::full::kInputToInputWeightsTensor); in lstm_eval()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/nnapi/
H A Dnnapi_delegate.cc5565 constexpr int kInputToInputWeightsTensor = 1; in AddOpsAndTensors() local
5579 if (input_pos == kInputToInputWeightsTensor || in AddOpsAndTensors()