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

  /external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/
lstm_utils.h 40 kCellToOutputWeightsTensor = 11, // Optional
identify_lstm_split_inputs.cc 120 &(lstm_cell_op->inputs[kCellToOutputWeightsTensor]),
  /frameworks/ml/nn/common/operations/
LSTM.h 79 static constexpr int kCellToOutputWeightsTensor = 11; // Optional
LSTM.cpp 56 cell_to_output_weights_ = GetInput(operation, operands, kCellToOutputWeightsTensor); // optional
161 GetInput(operation, operands, LSTMCell::kCellToOutputWeightsTensor);
LSTMTest.cpp 224 execution.setInput(LSTMCell::kCellToOutputWeightsTensor, nullptr, 0);
    [all...]
  /external/tensorflow/tensorflow/contrib/lite/kernels/
lstm.cc 54 constexpr int kCellToOutputWeightsTensor = 11; // Optional
154 GetOptionalInputTensor(context, node, kCellToOutputWeightsTensor);
333 GetOptionalInputTensor(context, node, kCellToOutputWeightsTensor);
unidirectional_sequence_lstm.cc 54 constexpr int kCellToOutputWeightsTensor = 11; // Optional
154 GetOptionalInputTensor(context, node, kCellToOutputWeightsTensor);
335 GetOptionalInputTensor(context, node, kCellToOutputWeightsTensor);

Completed in 520 milliseconds