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

  /external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/
lstm_utils.h 45 kProjectionWeightsTensor = 16, // Optional
identify_lstm_split_inputs.cc 136 CreateOptionalArray(model, &(lstm_cell_op->inputs[kProjectionWeightsTensor]),
  /frameworks/ml/nn/common/operations/
LSTM.h 88 static constexpr int kProjectionWeightsTensor = 16; // Optional
LSTM.cpp 63 projection_weights_ = GetInput(operation, operands, kProjectionWeightsTensor); // optional
204 GetInput(operation, operands, LSTMCell::kProjectionWeightsTensor);
LSTMTest.cpp 232 execution.setInput(LSTMCell::kProjectionWeightsTensor, nullptr, 0);
    [all...]
  /external/tensorflow/tensorflow/contrib/lite/kernels/
lstm.cc 63 constexpr int kProjectionWeightsTensor = 16; // Optional
196 GetOptionalInputTensor(context, node, kProjectionWeightsTensor);
344 GetOptionalInputTensor(context, node, kProjectionWeightsTensor);
unidirectional_sequence_lstm.cc 63 constexpr int kProjectionWeightsTensor = 16; // Optional
196 GetOptionalInputTensor(context, node, kProjectionWeightsTensor);
346 GetOptionalInputTensor(context, node, kProjectionWeightsTensor);

Completed in 171 milliseconds