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

  /external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/
lstm_utils.h 37 kRecurrentToOutputWeightsTensor = 8,
identify_lstm_merge_inputs.cc 112 model->GetArray(src_op->inputs[kRecurrentToOutputWeightsTensor]),
175 DeleteArrayIfUnused(src_op->inputs[kRecurrentToOutputWeightsTensor], model);
identify_lstm_split_inputs.cc 109 &(lstm_cell_op->inputs[kRecurrentToOutputWeightsTensor]),
  /frameworks/ml/nn/common/operations/
LSTM.h 74 static constexpr int kRecurrentToOutputWeightsTensor = 8;
LSTM.cpp 52 recurrent_to_output_weights_ = GetInput(operation, operands, kRecurrentToOutputWeightsTensor);
256 GetInput(operation, operands, LSTMCell::kRecurrentToOutputWeightsTensor);
  /external/tensorflow/tensorflow/contrib/lite/kernels/
lstm.cc 49 constexpr int kRecurrentToOutputWeightsTensor = 8;
243 GetInput(context, node, kRecurrentToOutputWeightsTensor);
326 GetInput(context, node, kRecurrentToOutputWeightsTensor);
unidirectional_sequence_lstm.cc 49 constexpr int kRecurrentToOutputWeightsTensor = 8;
244 GetInput(context, node, kRecurrentToOutputWeightsTensor);
328 GetInput(context, node, kRecurrentToOutputWeightsTensor);

Completed in 137 milliseconds