Searched defs:output_ptr_batch (Results 1 – 6 of 6) sorted by relevance
27 float* hidden_state_ptr_batch, float* output_ptr_batch) { in RnnBatchStep()43 float* hidden_state_ptr_batch, float* output_ptr_batch) { in RnnBatchStep()127 float* hidden_state_ptr_batch, float* output_ptr_batch) { in RnnBatchStep()150 float* hidden_state_ptr_batch, float* output_ptr_batch) { in RnnBatchStep()
334 float* output_ptr_batch = in EvalFloat() local352 float* output_ptr_batch = in EvalFloat() local377 float* output_ptr_batch = fw_output_offset + s * fw_output_step; in EvalFloat() local400 float* output_ptr_batch = bw_output_offset + s * bw_output_step; in EvalFloat() local496 float* output_ptr_batch = in EvalHybrid() local518 float* output_ptr_batch = in EvalHybrid() local548 float* output_ptr_batch = fw_output_offset + s * fw_output_step; in EvalHybrid() local574 float* output_ptr_batch = bw_output_offset + s * bw_output_step; in EvalHybrid() local
175 float* output_ptr_batch = output->data.f + s * num_units * batch_size; in EvalFloat() local191 float* output_ptr_batch = in EvalFloat() local255 float* output_ptr_batch = output->data.f + s * num_units * batch_size; in EvalHybrid() local273 float* output_ptr_batch = in EvalHybrid() local
161 float* output_ptr_batch = output->data.f; in EvalFloat() local193 float* output_ptr_batch = output->data.f; in EvalHybrid() local
74 float* output_ptr_batch = output->data.f + b * num_units; in ApplyTimeWeightsBiasAndActivation() local82 float* output_ptr_batch = output->data.f + b * num_units; in ApplyTimeWeightsBiasAndActivation() local
93 inline void LstmStepWithAuxInput( in LstmStepWithAuxInput()430 inline void LstmStepWithAuxInput( in LstmStepWithAuxInput()1283 float* output_ptr_batch = in EvalHybrid() local