Searched refs:kInputToInputWeightsTensor (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | identify_lstm_merge_inputs.cc | 60 int num_cell = model->GetArray(src_op->inputs[kInputToInputWeightsTensor]) in Run()
63 int num_input = model->GetArray(src_op->inputs[kInputToInputWeightsTensor]) in Run()
91 model->GetArray(src_op->inputs[kInputToInputWeightsTensor]), 0, 0); in Run()
|
| D | lstm_utils.h | 33 kInputToInputWeightsTensor = 1, // Optional enumerator
|
| D | identify_lstm_split_inputs.cc | 95 model, &(lstm_cell_op->inputs[kInputToInputWeightsTensor]), in Run()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | lstm_shared.h | 31 constexpr int kInputToInputWeightsTensor = 1; // Optional variable
|
| D | unidirectional_sequence_lstm.cc | 92 context, node, lstm::full::kInputToInputWeightsTensor); in PopulateQuantizedLstmParams8x8_16()
422 context, node, lstm::full::kInputToInputWeightsTensor); in CheckInputTensorDimensions()
719 context, node, lstm::full::kInputToInputWeightsTensor); in PopulatePrecomputedZPTimesWeightsWithBias()
960 context, node, lstm::full::kInputToInputWeightsTensor); in Prepare()
1217 context, node, lstm::full::kInputToInputWeightsTensor); in Eval()
|
| D | lstm.cc | 182 GetOptionalInputTensor(context, node, kInputToInputWeightsTensor); in PopulateQuantizedLstmParams8x8_16()
470 GetOptionalInputTensor(context, node, kInputToInputWeightsTensor); in PopulateQuantizedLstmParams8x8_8()
876 GetOptionalInputTensor(context, node, kInputToInputWeightsTensor); in CheckInputTensorDimensions()
1165 GetOptionalInputTensor(context, node, kInputToInputWeightsTensor); in PopulatePrecomputedZPTimesWeightsWithBias()
1416 GetOptionalInputTensor(context, node, kInputToInputWeightsTensor); in Prepare()
1599 GetOptionalInputTensor(context, node, kInputToInputWeightsTensor); in Prepare()
1629 GetOptionalInputTensor(context, node, kInputToInputWeightsTensor); in Prepare()
1833 GetOptionalInputTensor(context, node, kInputToInputWeightsTensor); in Eval()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | lstm_parser.cc | 76 node, tflite::ops::builtin::lstm::full::kInputToInputWeightsTensor); in HasCifg()
491 tflite::ops::builtin::lstm::full::kInputToInputWeightsTensor, in ParseLSTMAttributes()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| D | lstm.cc | 473 context, node, ops::builtin::lstm::full::kInputToInputWeightsTensor); in lstm_eval()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
| D | lstm.cc | 540 context, node, ops::builtin::lstm::full::kInputToInputWeightsTensor); in lstm_eval()
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate.cc | 4217 constexpr int kInputToInputWeightsTensor = 1; in AddOpsAndTensors() local
4231 if (input_pos == kInputToInputWeightsTensor || in AddOpsAndTensors()
|