HomeSort by relevance Sort by last modified time
    Searched defs:CheckInputTensorDimensions (Results 1 - 3 of 3) sorted by null

  /frameworks/ml/nn/common/operations/
LSTM.cpp 81 bool LSTMCell::CheckInputTensorDimensions(
263 if (!CheckInputTensorDimensions(operation, operands, n_input, n_output, n_cell)) {
  /external/tensorflow/tensorflow/contrib/lite/kernels/
lstm.cc 74 TfLiteStatus CheckInputTensorDimensions(TfLiteContext* context,
250 CheckInputTensorDimensions(context, node, n_input, n_output, n_cell);
unidirectional_sequence_lstm.cc 74 TfLiteStatus CheckInputTensorDimensions(TfLiteContext* context,
251 CheckInputTensorDimensions(context, node, n_input, n_output, n_cell);

Completed in 173 milliseconds