Searched refs:fw_recurrent_to_output_weights_ (Results 1 – 2 of 2) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | BidirectionalSequenceLSTM.cpp | 96 fw_recurrent_to_output_weights_ = in BidirectionalSequenceLSTM()
277 NN_CHECK_EQ(NumDimensions(fw_recurrent_to_output_weights_), 2); in Prepare()
278 NN_CHECK_EQ(SizeOfDimension(fw_recurrent_to_output_weights_, 0), n_fw_cell); in Prepare()
279 const uint32_t n_fw_output = SizeOfDimension(fw_recurrent_to_output_weights_, 1); in Prepare()
292 fw_recurrent_to_cell_weights_, fw_recurrent_to_output_weights_, in Prepare()
441 const uint32_t n_fw_output = SizeOfDimension(fw_recurrent_to_output_weights_, 1); in Eval()
495 GetBuffer<const float>(fw_recurrent_to_output_weights_), in Eval()
496 fw_recurrent_to_output_weights_->shape(), in Eval()
619 GetBuffer<const _Float16>(fw_recurrent_to_output_weights_), in Eval()
620 fw_recurrent_to_output_weights_->shape(), in Eval()
|
| D | BidirectionalSequenceLSTM.h | 183 const RunTimeOperandInfo* fw_recurrent_to_output_weights_; variable
|