Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DBidirectionalSequenceLSTM.cpp151 fw_aux_input_to_forget_weights_ = in BidirectionalSequenceLSTM()
308 !IsNullInput(fw_aux_input_to_forget_weights_) && in Prepare()
312 IsNullInput(fw_aux_input_to_forget_weights_) && in Prepare()
327 const bool has_fw_aux_weights = !IsNullInput(fw_aux_input_to_forget_weights_); in Prepare()
350 NN_RET_CHECK_EQ(NumDimensions(fw_aux_input_to_forget_weights_), 2); in Prepare()
351 NN_RET_CHECK_EQ(SizeOfDimension(fw_aux_input_to_forget_weights_, 0), n_fw_cell); in Prepare()
352 NN_RET_CHECK_EQ(SizeOfDimension(fw_aux_input_to_forget_weights_, 1), n_aux_input); in Prepare()
452 const bool has_aux_weights = !IsNullInput(fw_aux_input_to_forget_weights_); in Eval()
501 GetOptionalBuffer<const float>(fw_aux_input_to_forget_weights_), in Eval()
625 GetOptionalBuffer<const _Float16>(fw_aux_input_to_forget_weights_), in Eval()
DBidirectionalSequenceLSTM.h167 const RunTimeOperandInfo* fw_aux_input_to_forget_weights_; variable