Searched defs:CudnnBatchNormForwardInferenceParams (Results 1 – 1 of 1) sorted by relevance
39 struct CudnnBatchNormForwardInferenceParams { struct40 CudnnBatchNormParamsCommon common;41 se::DeviceMemoryBase output;42 se::DeviceMemory<float> offset;43 se::DeviceMemory<float> mean;44 se::DeviceMemory<float> variance;