Searched defs:CudnnBatchNormBackwardParams (Results 1 – 1 of 1) sorted by relevance
55 struct CudnnBatchNormBackwardParams { struct56 CudnnBatchNormParamsCommon common;57 se::DeviceMemoryBase output_grad_data;58 se::DeviceMemoryBase grad_output;59 se::DeviceMemory<float> output_grad_scale;60 se::DeviceMemory<float> output_grad_offset;61 se::DeviceMemory<float> mean;62 se::DeviceMemory<float> inv_stddev;