Home
last modified time | relevance | path

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

/packages/modules/NeuralNetworks/common/cpu_operations/
DBidirectionalSequenceLSTM.cpp155 bw_aux_input_to_input_weights_ = GetInput(operation, operands, kBwAuxInputToInputWeightsTensor); in BidirectionalSequenceLSTM()
302 IsNullInput(bw_aux_input_to_input_weights_)); in Prepare()
315 ((params_.use_cifg || !IsNullInput(bw_aux_input_to_input_weights_)) && in Prepare()
319 (IsNullInput(bw_aux_input_to_input_weights_) && in Prepare()
364 NN_RET_CHECK_EQ(NumDimensions(bw_aux_input_to_input_weights_), 2u); in Prepare()
365 NN_RET_CHECK_EQ(SizeOfDimension(bw_aux_input_to_input_weights_, 0), n_bw_cell); in Prepare()
366 NN_RET_CHECK_EQ(SizeOfDimension(bw_aux_input_to_input_weights_, 1), n_aux_input); in Prepare()
551 GetOptionalBuffer<const float>(bw_aux_input_to_input_weights_), in Eval()
676 GetOptionalBuffer<const _Float16>(bw_aux_input_to_input_weights_), in Eval()
/packages/modules/NeuralNetworks/common/types/operations/include/
DBidirectionalSequenceLSTM.h167 const RunTimeOperandInfo* bw_aux_input_to_input_weights_; variable