Searched refs:kHiddenStateInTensor (Results 1 – 2 of 2) sorted by relevance
| /frameworks/ml/nn/common/operations/ | ||
| D | RNN.h | 43 static constexpr int kHiddenStateInTensor = 4; variable |
| D | RNN.cpp | 39 hidden_state_in_ = GetInput(operation, operands, kHiddenStateInTensor); in RNN() |