Searched defs:CudnnBatchNormForwardTrainingParams (Results 1 – 1 of 1) sorted by relevance
47 struct CudnnBatchNormForwardTrainingParams { struct48 CudnnBatchNormParamsCommon common;49 se::DeviceMemoryBase output_data;50 se::DeviceMemory<float> offset;51 se::DeviceMemory<float> output_mean;52 se::DeviceMemory<float> output_inv_stddev;