Searched defs:scale_offset_descriptor (Results 1 – 2 of 2) sorted by relevance
1698 const CudnnTensorDescriptor& scale_offset_descriptor, in CreateBatchNormForwardWorkspace()1721 const CudnnTensorDescriptor& scale_offset_descriptor, in CreateBatchNormBackwardWorkspace()3509 CudnnTensorDescriptor scale_offset_descriptor( in DoBatchNormalizationForwardImpl() local3690 CudnnTensorDescriptor scale_offset_descriptor( in DoBatchNormalizationBackwardImpl() local
3579 ScopedTensorDescriptor scale_offset_descriptor{ in DoBatchNormalizationForwardImpl() local3654 ScopedTensorDescriptor scale_offset_descriptor{ in DoBatchNormalizationBackwardImpl() local