Searched defs:activation_desc (Results 1 – 2 of 2) sorted by relevance
1877 const cudnnActivationDescriptor_t& activation_desc, in CreateBatchNormForwardWorkspace()1901 const cudnnActivationDescriptor_t& activation_desc, in CreateBatchNormBackwardWorkspace()4886 CudnnActivationDescriptor activation_desc) { in Create()5024 CudnnActivationDescriptor activation_desc) in CudnnLegacyFusedConvRunner()5096 CudnnActivationDescriptor activation_desc(activation_mode, in FusedConvolveRunnerFromDesc() local5497 CudnnActivationDescriptor activation_desc( in DoBatchNormalizationForwardImpl() local5688 CudnnActivationDescriptor activation_desc( in DoBatchNormalizationBackwardImpl() local5821 CudnnActivationDescriptor activation_desc( in DoFusedConvolve() local6141 CudnnActivationDescriptor activation_desc( in DoActivate() local
4584 ScopedActivationDescriptor activation_desc{activation_mode}; in DoFusedConvolutionBiasActivationImpl() local4682 ScopedActivationDescriptor activation_desc{activation_mode}; in DoFusedBatchNormActivationInferenceImpl() local4793 ScopedActivationDescriptor activation_desc{activation_mode}; in DoFusedBatchNormActivationForwardImpl() local4909 ScopedActivationDescriptor activation_desc{activation_mode}; in DoFusedBatchNormActivationBackwardImpl() local