Searched refs:kInputToInputWeightsTensor (Results 1 – 5 of 5) 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()
90 model->GetArray(src_op->inputs[kInputToInputWeightsTensor]), 0, 0); in Run()
176 DeleteArrayIfUnused(src_op->inputs[kInputToInputWeightsTensor], model); 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 | unidirectional_sequence_lstm.cc | 41 constexpr int kInputToInputWeightsTensor = 1; // Optional variable
112 GetOptionalInputTensor(context, node, kInputToInputWeightsTensor); in CheckInputTensorDimensions()
327 GetOptionalInputTensor(context, node, kInputToInputWeightsTensor); in Prepare()
445 GetOptionalInputTensor(context, node, kInputToInputWeightsTensor); in Eval()
|
| D | lstm.cc | 65 constexpr int kInputToInputWeightsTensor = 1; // Optional variable
129 GetOptionalInputTensor(context, node, kInputToInputWeightsTensor); in CheckInputTensorDimensions()
403 GetOptionalInputTensor(context, node, kInputToInputWeightsTensor); in Prepare()
513 GetOptionalInputTensor(context, node, kInputToInputWeightsTensor); in Eval()
|