Home
last modified time | relevance | path

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

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DBidirectionalSequenceLSTM.cpp159 bw_aux_input_to_output_weights_ = in BidirectionalSequenceLSTM()
318 !IsNullInput(bw_aux_input_to_output_weights_)) || in Prepare()
322 IsNullInput(bw_aux_input_to_output_weights_)); in Prepare()
376 NN_RET_CHECK_EQ(NumDimensions(bw_aux_input_to_output_weights_), 2); in Prepare()
377 NN_RET_CHECK_EQ(SizeOfDimension(bw_aux_input_to_output_weights_, 0), n_bw_cell); in Prepare()
378 NN_RET_CHECK_EQ(SizeOfDimension(bw_aux_input_to_output_weights_, 1), n_aux_input); in Prepare()
554 GetOptionalBuffer<const float>(bw_aux_input_to_output_weights_), in Eval()
679 GetOptionalBuffer<const _Float16>(bw_aux_input_to_output_weights_), in Eval()
H A DBidirectionalSequenceLSTM.h170 const RunTimeOperandInfo* bw_aux_input_to_output_weights_; variable