336         NN_CHECK_EQ(aux_input_->shape().dimensions[0], input_->shape().dimensions[0]);  in Prepare()
337         NN_CHECK_EQ(aux_input_->shape().dimensions[1], input_->shape().dimensions[1]);  in Prepare()
382     const Shape& inputShape = input_->shape();  in Prepare()
419         *fwOutputActivationState = fw_activation_state_->shape();  in Prepare()
420         *fwOutputCellState = fw_cell_state_->shape();  in Prepare()
421         *bwOutputActivationState = bw_activation_state_->shape();  in Prepare()
422         *bwOutputCellState = bw_cell_state_->shape();  in Prepare()
442     std::vector<uint32_t> fw_output_dims = input_->shape().dimensions;  in Eval()
459             Shape bwInputShape = input_->shape();  in Eval()
463                 bwInputShape = aux_input_->shape();  in Eval()
476                         getNumberOfElements(fw_activation_state_->shape()));  in Eval()
477                 fw_output_cell_state.resize(getNumberOfElements(fw_cell_state_->shape()));  in Eval()
485                     params_, GetBuffer<const float>(input_), input_->shape(),  in Eval()
490                     fw_input_to_output_weights_->shape(),  in Eval()
495                     fw_recurrent_to_output_weights_->shape(),  in Eval()
528                         getNumberOfElements(bw_activation_state_->shape()));  in Eval()
529                 bw_output_cell_state.resize(getNumberOfElements(bw_cell_state_->shape()));  in Eval()
542                     bw_input_to_output_weights_->shape(),  in Eval()
547                     bw_recurrent_to_output_weights_->shape(),  in Eval()
582             Shape bwInputShape = input_->shape();  in Eval()
586                 bwInputShape = aux_input_->shape();  in Eval()
600                         getNumberOfElements(fw_activation_state_->shape()));  in Eval()
601                 fw_output_cell_state.resize(getNumberOfElements(fw_cell_state_->shape()));  in Eval()
609                     params_, GetBuffer<const _Float16>(input_), input_->shape(),  in Eval()
614                     fw_input_to_output_weights_->shape(),  in Eval()
619                     fw_recurrent_to_output_weights_->shape(),  in Eval()
653                         getNumberOfElements(bw_activation_state_->shape()));  in Eval()
654                 bw_output_cell_state.resize(getNumberOfElements(bw_cell_state_->shape()));  in Eval()
667                     bw_input_to_output_weights_->shape(),  in Eval()
672                     bw_recurrent_to_output_weights_->shape(),  in Eval()