Searched refs:fw_aux_input_to_input_weights_ (Results 1 – 2 of 2) sorted by relevance
149 fw_aux_input_to_input_weights_ = GetInput(operation, operands, kFwAuxInputToInputWeightsTensor); in BidirectionalSequenceLSTM()301 NN_RET_CHECK(IsNullInput(fw_aux_input_to_input_weights_) && in Prepare()306 ((params_.use_cifg || !IsNullInput(fw_aux_input_to_input_weights_)) && in Prepare()310 (IsNullInput(fw_aux_input_to_input_weights_) && in Prepare()345 NN_RET_CHECK_EQ(NumDimensions(fw_aux_input_to_input_weights_), 2u); in Prepare()346 NN_RET_CHECK_EQ(SizeOfDimension(fw_aux_input_to_input_weights_, 0), n_fw_cell); in Prepare()347 NN_RET_CHECK_EQ(SizeOfDimension(fw_aux_input_to_input_weights_, 1), n_aux_input); in Prepare()499 GetOptionalBuffer<const float>(fw_aux_input_to_input_weights_), in Eval()623 GetOptionalBuffer<const _Float16>(fw_aux_input_to_input_weights_), in Eval()
163 const RunTimeOperandInfo* fw_aux_input_to_input_weights_; variable