Searched refs:fw_aux_input_to_input_weights_ (Results 1 – 2 of 2) sorted by relevance
150 fw_aux_input_to_input_weights_ = GetInput(operation, operands, kFwAuxInputToInputWeightsTensor); in BidirectionalSequenceLSTM()302 NN_RET_CHECK(IsNullInput(fw_aux_input_to_input_weights_) && in Prepare()307 ((params_.use_cifg || !IsNullInput(fw_aux_input_to_input_weights_)) && in Prepare()311 (IsNullInput(fw_aux_input_to_input_weights_) && in Prepare()346 NN_RET_CHECK_EQ(NumDimensions(fw_aux_input_to_input_weights_), 2); in Prepare()347 NN_RET_CHECK_EQ(SizeOfDimension(fw_aux_input_to_input_weights_, 0), n_fw_cell); in Prepare()348 NN_RET_CHECK_EQ(SizeOfDimension(fw_aux_input_to_input_weights_, 1), n_aux_input); in Prepare()500 GetOptionalBuffer<const float>(fw_aux_input_to_input_weights_), in Eval()624 GetOptionalBuffer<const _Float16>(fw_aux_input_to_input_weights_), in Eval()
166 const RunTimeOperandInfo* fw_aux_input_to_input_weights_; variable