Home
last modified time | relevance | path

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

/packages/modules/NeuralNetworks/common/cpu_operations/
DBidirectionalSequenceLSTM.cpp156 bw_aux_input_to_forget_weights_ = in BidirectionalSequenceLSTM()
316 !IsNullInput(bw_aux_input_to_forget_weights_) && in Prepare()
320 IsNullInput(bw_aux_input_to_forget_weights_) && in Prepare()
327 const bool has_bw_aux_weights = !IsNullInput(bw_aux_input_to_forget_weights_); in Prepare()
368 NN_RET_CHECK_EQ(NumDimensions(bw_aux_input_to_forget_weights_), 2u); in Prepare()
369 NN_RET_CHECK_EQ(SizeOfDimension(bw_aux_input_to_forget_weights_, 0), n_bw_cell); in Prepare()
370 NN_RET_CHECK_EQ(SizeOfDimension(bw_aux_input_to_forget_weights_, 1), n_aux_input); in Prepare()
552 GetOptionalBuffer<const float>(bw_aux_input_to_forget_weights_), in Eval()
677 GetOptionalBuffer<const _Float16>(bw_aux_input_to_forget_weights_), in Eval()
/packages/modules/NeuralNetworks/common/types/operations/include/
DBidirectionalSequenceLSTM.h168 const RunTimeOperandInfo* bw_aux_input_to_forget_weights_; variable