HomeSort by relevance Sort by last modified time
    Searched refs:kInputToInputWeightsTensor (Results 1 - 5 of 5) sorted by null

  /external/tensorflow/tensorflow/lite/toco/graph_transformations/
identify_lstm_merge_inputs.cc 60 int num_cell = model->GetArray(src_op->inputs[kInputToInputWeightsTensor])
63 int num_input = model->GetArray(src_op->inputs[kInputToInputWeightsTensor])
90 model->GetArray(src_op->inputs[kInputToInputWeightsTensor]), 0, 0);
176 DeleteArrayIfUnused(src_op->inputs[kInputToInputWeightsTensor], model);
lstm_utils.h 33 kInputToInputWeightsTensor = 1, // Optional
identify_lstm_split_inputs.cc 95 model, &(lstm_cell_op->inputs[kInputToInputWeightsTensor]),
  /external/tensorflow/tensorflow/lite/kernels/
unidirectional_sequence_lstm.cc 41 constexpr int kInputToInputWeightsTensor = 1; // Optional
112 GetOptionalInputTensor(context, node, kInputToInputWeightsTensor);
327 GetOptionalInputTensor(context, node, kInputToInputWeightsTensor);
445 GetOptionalInputTensor(context, node, kInputToInputWeightsTensor);
lstm.cc 65 constexpr int kInputToInputWeightsTensor = 1; // Optional
129 GetOptionalInputTensor(context, node, kInputToInputWeightsTensor);
403 GetOptionalInputTensor(context, node, kInputToInputWeightsTensor);
513 GetOptionalInputTensor(context, node, kInputToInputWeightsTensor);
    [all...]

Completed in 1088 milliseconds