Searched refs:forward_params (Results 1 – 1 of 1) sorted by relevance
260 CudnnBatchNormForwardTrainingParams forward_params; in RunCudnnBatchNormForwardTraining() local261 AssignCommonParams(config, &forward_params.common, operand, scale); in RunCudnnBatchNormForwardTraining()262 forward_params.offset = offset; in RunCudnnBatchNormForwardTraining()263 forward_params.output_data = output_data; in RunCudnnBatchNormForwardTraining()264 forward_params.output_mean = output_mean; in RunCudnnBatchNormForwardTraining()265 forward_params.output_inv_stddev = output_inv_stddev; in RunCudnnBatchNormForwardTraining()269 RunCudnnBatchNormForwardTrainingImpl<Eigen::half>(&forward_params, in RunCudnnBatchNormForwardTraining()273 RunCudnnBatchNormForwardTrainingImpl<float>(&forward_params, stream); in RunCudnnBatchNormForwardTraining()