Searched refs:ScopedActivationDescriptor (Results 1 – 1 of 1) sorted by relevance
819 class ScopedActivationDescriptor { class821 ScopedActivationDescriptor(dnn::ActivationMode activation_mode) in ScopedActivationDescriptor() function in stream_executor::gpu::ScopedActivationDescriptor870 ~ScopedActivationDescriptor() { in ~ScopedActivationDescriptor()895 SE_DISALLOW_COPY_AND_ASSIGN(ScopedActivationDescriptor);1122 ScopedActivationDescriptor& activation_descriptor) in ScopedFusionPlanConvolutionBiasActivation()1188 ScopedActivationDescriptor& activation_descriptor) { in SetActivationForwardArgs()1202 ScopedActivationDescriptor& activation_descriptor) { in GetFusionOpHashValue()1236 ScopedActivationDescriptor& activation_descriptor) in ScopedFusionPlanBatchNormActivationInference()1294 ScopedActivationDescriptor& activation_descriptor) { in SetActivationForwardArgs()1306 ScopedActivationDescriptor& activation_descriptor) { in GetFusionOpHashValue()[all …]