Searched defs:output_ptr_batch (Results 1 – 6 of 6) sorted by relevance
430 float* output_ptr_batch = in EvalFloat() local448 float* output_ptr_batch = in EvalFloat() local476 float* output_ptr_batch = fw_output_offset + s * fw_output_step; in EvalFloat() local502 float* output_ptr_batch = bw_output_offset + s * bw_output_step; in EvalFloat() local603 float* output_ptr_batch = in EvalHybrid() local627 float* output_ptr_batch = in EvalHybrid() local662 float* output_ptr_batch = fw_output_offset + s * fw_output_step; in EvalHybrid() local693 float* output_ptr_batch = bw_output_offset + s * bw_output_step; in EvalHybrid() local
236 float* output_ptr_batch = in EvalFloat() local255 float* output_ptr_batch = GetTensorData<float>(output) + in EvalFloat() local317 float* output_ptr_batch = in EvalHybrid() local340 float* output_ptr_batch = GetTensorData<float>(output) + in EvalHybrid() local
216 float* output_ptr_batch = GetTensorData<float>(output); in EvalFloat() local249 float* output_ptr_batch = GetTensorData<float>(output); in EvalHybrid() local
29 float* hidden_state_ptr_batch, float* output_ptr_batch) { in RnnBatchStep()45 float* hidden_state_ptr_batch, float* output_ptr_batch) { in RnnBatchStep()126 float* hidden_state_ptr_batch, float* output_ptr_batch, in RnnBatchStep()153 float* hidden_state_ptr_batch, float* output_ptr_batch, in RnnBatchStep()
146 auto* output_ptr_batch = GetBuffer<int32_t>(output_) + b * sample_count_; in EvalFloat32() local
168 T* output_ptr_batch = outputData + b * outputBatchStride + outputBatchOffset; in RNNStep() local