Searched defs:CudnnBatchNormParamsCommon (Results 1 – 1 of 1) sorted by relevance
31 struct CudnnBatchNormParamsCommon { struct32 se::DeviceMemoryBase operand;33 se::dnn::BatchDescriptor operand_desc;34 se::dnn::BatchDescriptor scale_offset_desc;35 se::DeviceMemory<float> scale;36 float epsilon;