Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DBidirectionalSequenceLSTM.cpp160 bw_aux_input_to_output_weights_ = in BidirectionalSequenceLSTM()
319 !IsNullInput(bw_aux_input_to_output_weights_)) || in Prepare()
323 IsNullInput(bw_aux_input_to_output_weights_)); in Prepare()
377 NN_RET_CHECK_EQ(NumDimensions(bw_aux_input_to_output_weights_), 2); in Prepare()
378 NN_RET_CHECK_EQ(SizeOfDimension(bw_aux_input_to_output_weights_, 0), n_bw_cell); in Prepare()
379 NN_RET_CHECK_EQ(SizeOfDimension(bw_aux_input_to_output_weights_, 1), n_aux_input); in Prepare()
555 GetOptionalBuffer<const float>(bw_aux_input_to_output_weights_), in Eval()
680 GetOptionalBuffer<const _Float16>(bw_aux_input_to_output_weights_), in Eval()
DBidirectionalSequenceLSTM.h173 const RunTimeOperandInfo* bw_aux_input_to_output_weights_; variable