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

  /external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/
identify_lstm_merge_inputs.cc 56 int num_cell = model->GetArray(src_op->inputs[kInputToInputWeightsTensor])
59 int num_input = model->GetArray(src_op->inputs[kInputToInputWeightsTensor])
86 model->GetArray(src_op->inputs[kInputToInputWeightsTensor]), 0, 0);
168 DeleteArrayIfUnused(src_op->inputs[kInputToInputWeightsTensor], model);
lstm_utils.h 30 kInputToInputWeightsTensor = 1, // Optional
identify_lstm_split_inputs.cc 84 model, &(lstm_cell_op->inputs[kInputToInputWeightsTensor]),
  /frameworks/ml/nn/common/operations/
LSTM.h 65 static constexpr int kInputToInputWeightsTensor = 1; // Optional
LSTM.cpp 43 input_to_input_weights_ = GetInput(operation, operands, kInputToInputWeightsTensor); // optional
98 GetInput(operation, operands, LSTMCell::kInputToInputWeightsTensor);
286 GetInput(operation, operands, LSTMCell::kInputToInputWeightsTensor);
LSTMTest.cpp 213 execution.setInput(LSTMCell::kInputToInputWeightsTensor, nullptr, 0);
    [all...]
  /external/tensorflow/tensorflow/contrib/lite/kernels/
lstm.cc 40 constexpr int kInputToInputWeightsTensor = 1; // Optional
86 GetOptionalInputTensor(context, node, kInputToInputWeightsTensor);
285 GetOptionalInputTensor(context, node, kInputToInputWeightsTensor);
311 GetOptionalInputTensor(context, node, kInputToInputWeightsTensor);
unidirectional_sequence_lstm.cc 40 constexpr int kInputToInputWeightsTensor = 1; // Optional
86 GetOptionalInputTensor(context, node, kInputToInputWeightsTensor);
287 GetOptionalInputTensor(context, node, kInputToInputWeightsTensor);
313 GetOptionalInputTensor(context, node, kInputToInputWeightsTensor);

Completed in 69 milliseconds