Searched refs:bw_recurrent_to_output_weights_ (Results 1 – 2 of 2) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | BidirectionalSequenceLSTM.cpp | 128 bw_recurrent_to_output_weights_ = in BidirectionalSequenceLSTM()
283 NN_CHECK_EQ(NumDimensions(bw_recurrent_to_output_weights_), 2); in Prepare()
284 NN_CHECK_EQ(SizeOfDimension(bw_recurrent_to_output_weights_, 0), n_bw_cell); in Prepare()
285 const uint32_t n_bw_output = SizeOfDimension(bw_recurrent_to_output_weights_, 1); in Prepare()
400 bw_recurrent_to_cell_weights_, bw_recurrent_to_output_weights_, in Prepare()
442 const uint32_t n_bw_output = SizeOfDimension(bw_recurrent_to_output_weights_, 1); in Eval()
547 GetBuffer<const float>(bw_recurrent_to_output_weights_), in Eval()
548 bw_recurrent_to_output_weights_->shape(), in Eval()
672 GetBuffer<const _Float16>(bw_recurrent_to_output_weights_), in Eval()
673 bw_recurrent_to_output_weights_->shape(), in Eval()
|
| D | BidirectionalSequenceLSTM.h | 214 const RunTimeOperandInfo* bw_recurrent_to_output_weights_; variable
|