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

  /external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/
lstm_utils.h 38 kCellToInputWeightsTensor = 9, // Optional
identify_lstm_split_inputs.cc 114 CreateOptionalArray(model, &(lstm_cell_op->inputs[kCellToInputWeightsTensor]),
  /frameworks/ml/nn/common/operations/
LSTM.h 77 static constexpr int kCellToInputWeightsTensor = 9; // Optional
LSTM.cpp 54 cell_to_input_weights_ = GetInput(operation, operands, kCellToInputWeightsTensor); // optional
147 GetInput(operation, operands, LSTMCell::kCellToInputWeightsTensor);
LSTMTest.cpp 219 execution.setInput(LSTMCell::kCellToInputWeightsTensor, nullptr, 0);
222 execution.setInput(LSTMCell::kCellToInputWeightsTensor, nullptr, 0);
    [all...]
  /external/tensorflow/tensorflow/contrib/lite/kernels/
lstm.cc 52 constexpr int kCellToInputWeightsTensor = 9; // Optional
140 GetOptionalInputTensor(context, node, kCellToInputWeightsTensor);
329 GetOptionalInputTensor(context, node, kCellToInputWeightsTensor);
unidirectional_sequence_lstm.cc 52 constexpr int kCellToInputWeightsTensor = 9; // Optional
140 GetOptionalInputTensor(context, node, kCellToInputWeightsTensor);
331 GetOptionalInputTensor(context, node, kCellToInputWeightsTensor);

Completed in 201 milliseconds