Searched refs:output_ptr_batch (Results 1 – 3 of 3) sorted by relevance
172 T* output_ptr_batch = outputData + b * outputBatchStride + outputBatchOffset; in RNNStep() local184 output_ptr_batch[o] = biasData[o]; in RNNStep()190 output_ptr_batch[o] += input_ptr_batch[i] * input_weights_ptr[i]; in RNNStep()199 output_ptr_batch[o] += aux_input_ptr_batch[i] * aux_input_weights_ptr[i]; in RNNStep()208 output_ptr_batch[o] += hidden_state_in_ptr_batch[h] * recurrent_weights_ptr[h]; in RNNStep()215 output_ptr_batch[o] = in RNNStep()216 (ActivationFunctor(static_cast<ActivationFn>(activation)))(output_ptr_batch[o]); in RNNStep()218 *hiddenStateOutput = output_ptr_batch[o]; in RNNStep()
214 float* output_ptr_batch = outputData + b * num_units; in EvalFloat32() local216 tflite::tensor_utils::ReductionSumVector(scratch_ptr_batch, output_ptr_batch, num_units, in EvalFloat32()222 float* output_ptr_batch = outputData + b * num_units; in EvalFloat32() local223 tflite::tensor_utils::ApplyActivationToVector(output_ptr_batch, num_units, in EvalFloat32()224 params_.activation_, output_ptr_batch); in EvalFloat32()
140 auto* output_ptr_batch = GetBuffer<int32_t>(output_) + b * sample_count_; in EvalFloat32() local144 output_ptr_batch[j] = std::distance(cdf.begin(), found_iter); in EvalFloat32()