Searched refs:output_ptr_batch (Results 1 – 2 of 2) sorted by relevance
168 T* output_ptr_batch = outputData + b * outputBatchStride + outputBatchOffset; in RNNStep() local180 output_ptr_batch[o] = biasData[o]; in RNNStep()186 output_ptr_batch[o] += input_ptr_batch[i] * input_weights_ptr[i]; in RNNStep()195 output_ptr_batch[o] += aux_input_ptr_batch[i] * aux_input_weights_ptr[i]; in RNNStep()204 output_ptr_batch[o] += hidden_state_in_ptr_batch[h] * recurrent_weights_ptr[h]; in RNNStep()211 output_ptr_batch[o] = in RNNStep()212 (ActivationFunctor(static_cast<ActivationFn>(activation)))(output_ptr_batch[o]); in RNNStep()214 *hiddenStateOutput = output_ptr_batch[o]; in RNNStep()
146 auto* output_ptr_batch = GetBuffer<int32_t>(output_) + b * sample_count_; in EvalFloat32() local150 output_ptr_batch[j] = std::distance(cdf.begin(), found_iter); in EvalFloat32()