Searched defs:ActivationDescriptor (Results 1 – 6 of 6) sorted by relevance
36 struct ActivationDescriptor : BaseDescriptor struct38 ActivationDescriptor() in ActivationDescriptor() function44 ActivationDescriptor(armnn::ActivationFunction activation, in ActivationDescriptor() argument52 bool operator ==(const ActivationDescriptor &rhs) const in operator ==()59 ActivationFunction m_Function;61 float m_A;63 float m_B;
378 struct TORCH_CUDA_CPP_API ActivationDescriptor struct387 AT_CUDNN_CHECK(cudnnSetActivationDescriptor( in set() argument
60 struct ActivationDescriptor struct68 bool operator ==(const ActivationDescriptor &rhs) const; argument
182 const ActivationDescriptor& /*descriptor*/, in IsActivationSupported()
87 const ActivationDescriptor&, // descriptor in IsActivationSupported()
536 using ActivationDescriptor = typedef