Searched refs:kInputToInputWeightsTensor (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/ |
| D | identify_lstm_merge_inputs.cc | 56 int num_cell = model->GetArray(src_op->inputs[kInputToInputWeightsTensor]) in Run()
59 int num_input = model->GetArray(src_op->inputs[kInputToInputWeightsTensor]) in Run()
86 model->GetArray(src_op->inputs[kInputToInputWeightsTensor]), 0, 0); in Run()
168 DeleteArrayIfUnused(src_op->inputs[kInputToInputWeightsTensor], model); in Run()
|
| D | lstm_utils.h | 30 kInputToInputWeightsTensor = 1, // Optional enumerator
|
| D | identify_lstm_split_inputs.cc | 84 model, &(lstm_cell_op->inputs[kInputToInputWeightsTensor]), in Run()
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/ |
| D | lstm.cc | 40 constexpr int kInputToInputWeightsTensor = 1; // Optional variable
86 GetOptionalInputTensor(context, node, kInputToInputWeightsTensor); in CheckInputTensorDimensions()
285 GetOptionalInputTensor(context, node, kInputToInputWeightsTensor); in Prepare()
311 GetOptionalInputTensor(context, node, kInputToInputWeightsTensor); in Eval()
|
| D | unidirectional_sequence_lstm.cc | 40 constexpr int kInputToInputWeightsTensor = 1; // Optional variable
86 GetOptionalInputTensor(context, node, kInputToInputWeightsTensor); in CheckInputTensorDimensions()
287 GetOptionalInputTensor(context, node, kInputToInputWeightsTensor); in Prepare()
313 GetOptionalInputTensor(context, node, kInputToInputWeightsTensor); in Eval()
|