Searched refs:bw_input_to_output_weights_ (Results 1 – 2 of 2) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | BidirectionalSequenceLSTM.cpp | 88 bw_input_to_output_weights_ = GetInput(operation, operands, kBwInputToOutputWeightsTensor); in BidirectionalSequenceLSTM()
212 const uint32_t n_bw_cell = SizeOfDimension(bw_input_to_output_weights_, 0); in Prepare()
213 NN_CHECK_EQ(NumDimensions(bw_input_to_output_weights_), 2); in Prepare()
214 NN_CHECK_EQ(SizeOfDimension(bw_input_to_output_weights_, 1), n_input); in Prepare()
232 bw_input_to_cell_weights_, bw_input_to_output_weights_, in Prepare()
325 GetBuffer<const float>(bw_input_to_output_weights_), in Eval()
326 bw_input_to_output_weights_->shape(), in Eval()
411 GetBuffer<const _Float16>(bw_input_to_output_weights_), in Eval()
412 bw_input_to_output_weights_->shape(), in Eval()
|
| D | BidirectionalSequenceLSTM.h | 202 const RunTimeOperandInfo* bw_input_to_output_weights_; variable
|