Searched refs:fw_input_to_output_weights_ (Results 1 – 2 of 2) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | BidirectionalSequenceLSTM.cpp | 89 fw_input_to_output_weights_ = GetInput(operation, operands, kFwInputToOutputWeightsTensor); in BidirectionalSequenceLSTM()
273 const uint32_t n_fw_cell = SizeOfDimension(fw_input_to_output_weights_, 0); in Prepare()
274 NN_CHECK_EQ(NumDimensions(fw_input_to_output_weights_), 2); in Prepare()
275 NN_CHECK_EQ(SizeOfDimension(fw_input_to_output_weights_, 1), n_fw_input); in Prepare()
290 fw_input_to_cell_weights_, fw_input_to_output_weights_, in Prepare()
490 GetBuffer<const float>(fw_input_to_output_weights_), in Eval()
491 fw_input_to_output_weights_->shape(), in Eval()
614 GetBuffer<const _Float16>(fw_input_to_output_weights_), in Eval()
615 fw_input_to_output_weights_->shape(), in Eval()
|
| D | BidirectionalSequenceLSTM.h | 178 const RunTimeOperandInfo* fw_input_to_output_weights_; variable
|