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

  /frameworks/ml/nn/common/operations/
RNN.h 45 Shape *hiddenStateShape,
RNN.cpp 42 Shape *hiddenStateShape,
70 hiddenStateShape->type = inputShape.type;
71 hiddenStateShape->dimensions = { batch_size, num_units };
  /frameworks/ml/nn/common/
CpuExecutor.cpp     [all...]

Completed in 75 milliseconds