Searched refs:activation_desc (Results 1 – 2 of 2) sorted by relevance
4049 ScopedActivationDescriptor activation_desc{activation_mode}; in DoFusedConvolutionBiasActivationImpl() local4053 conv.handle(), bias_nd.handle(), activation_desc}; in DoFusedConvolutionBiasActivationImpl()4078 status = fusion_plan.SetActivationForwardArgs(activation_desc); in DoFusedConvolutionBiasActivationImpl()4147 ScopedActivationDescriptor activation_desc{activation_mode}; in DoFusedBatchNormActivationInferenceImpl() local4151 activation_desc}; in DoFusedBatchNormActivationInferenceImpl()4174 status = fusion_plan.SetActivationForwardArgs(activation_desc); in DoFusedBatchNormActivationInferenceImpl()4258 ScopedActivationDescriptor activation_desc{activation_mode}; in DoFusedBatchNormActivationForwardImpl() local4262 activation_desc}; in DoFusedBatchNormActivationForwardImpl()4286 status = fusion_plan.SetActivationForwardArgs(activation_desc); in DoFusedBatchNormActivationForwardImpl()4374 ScopedActivationDescriptor activation_desc{activation_mode}; in DoFusedBatchNormActivationBackwardImpl() local[all …]
2981 CudnnActivationDescriptor activation_desc( in DoFusedConvolveImpl() local3000 << "\nactivation_desc.handle() = " << activation_desc.handle() in DoFusedConvolveImpl()3022 /*activationDesc=*/activation_desc.handle(), in DoFusedConvolveImpl()3709 CudnnActivationDescriptor activation_desc( in DoActivate() local3721 cudnn.handle(), activation_desc.handle(), &alpha, input_nd.handle(), in DoActivate()