Searched defs:scale_offset_descriptor (Results 1 – 2 of 2) sorted by relevance
1879 const CudnnTensorDescriptor& scale_offset_descriptor, in CreateBatchNormForwardWorkspace()1903 const CudnnTensorDescriptor& scale_offset_descriptor, in CreateBatchNormBackwardWorkspace()5471 CudnnTensorDescriptor scale_offset_descriptor( in DoBatchNormalizationForwardImpl() local5661 CudnnTensorDescriptor scale_offset_descriptor( in DoBatchNormalizationBackwardImpl() local
3695 ScopedTensorDescriptor scale_offset_descriptor{ in DoBatchNormalizationForwardImpl() local3774 ScopedTensorDescriptor scale_offset_descriptor{ in DoBatchNormalizationBackwardImpl() local