Searched refs:backward_params (Results 1 – 1 of 1) sorted by relevance
291 CudnnBatchNormBackwardParams backward_params; in RunCudnnBatchNormBackward() local292 AssignCommonParams(config, &backward_params.common, operand, scale); in RunCudnnBatchNormBackward()293 backward_params.output_grad_data = output_grad_data; in RunCudnnBatchNormBackward()294 backward_params.grad_output = grad_output; in RunCudnnBatchNormBackward()295 backward_params.output_grad_scale = output_grad_scale; in RunCudnnBatchNormBackward()296 backward_params.output_grad_offset = output_grad_offset; in RunCudnnBatchNormBackward()297 backward_params.mean = mean; in RunCudnnBatchNormBackward()298 backward_params.inv_stddev = inv_stddev; in RunCudnnBatchNormBackward()302 RunCudnnBatchNormBackwardImpl<Eigen::half>(&backward_params, stream); in RunCudnnBatchNormBackward()305 RunCudnnBatchNormBackwardImpl<float>(&backward_params, stream); in RunCudnnBatchNormBackward()