Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DBidirectionalSequenceLSTM.cpp156 bw_aux_input_to_input_weights_ = GetInput(operation, operands, kBwAuxInputToInputWeightsTensor); in BidirectionalSequenceLSTM()
303 IsNullInput(bw_aux_input_to_input_weights_)); in Prepare()
316 ((params_.use_cifg || !IsNullInput(bw_aux_input_to_input_weights_)) && in Prepare()
320 (IsNullInput(bw_aux_input_to_input_weights_) && in Prepare()
365 NN_RET_CHECK_EQ(NumDimensions(bw_aux_input_to_input_weights_), 2); in Prepare()
366 NN_RET_CHECK_EQ(SizeOfDimension(bw_aux_input_to_input_weights_, 0), n_bw_cell); in Prepare()
367 NN_RET_CHECK_EQ(SizeOfDimension(bw_aux_input_to_input_weights_, 1), n_aux_input); in Prepare()
552 GetOptionalBuffer<const float>(bw_aux_input_to_input_weights_), in Eval()
677 GetOptionalBuffer<const _Float16>(bw_aux_input_to_input_weights_), in Eval()
DBidirectionalSequenceLSTM.h170 const RunTimeOperandInfo* bw_aux_input_to_input_weights_; variable