Home
last modified time | relevance | path

Searched refs:recurrent_weights_ (Results 1 – 2 of 2) sorted by relevance

/frameworks/ml/nn/common/operations/
DRNN.cpp38 recurrent_weights_ = GetInput(operation, operands, kRecurrentWeightsTensor); in RNN()
92 reinterpret_cast<_Float16*>(recurrent_weights_->buffer), in Eval()
93 recurrent_weights_->shape(), activation_, in Eval()
104 reinterpret_cast<float*>(recurrent_weights_->buffer), in Eval()
105 recurrent_weights_->shape(), activation_, in Eval()
DRNN.h69 const RunTimeOperandInfo* recurrent_weights_; variable