Searched refs:activation_type (Results 1 – 3 of 3) sorted by relevance
42 attr->activation_type = static_cast<mindspore::schema::ActivationType>(addAttr.activationType); in GetAddPrimitive()71 …attr->activation_type = static_cast<mindspore::schema::ActivationType>(avgPoolAttr.activationType); in GetAvgPoolPrimitive()124 … attr->activation_type = static_cast<mindspore::schema::ActivationType>(conv2dAttr.activationType); in GetConv2dPrimitive()151 …attr->activation_type = static_cast<mindspore::schema::ActivationType>(fullConnAttr.activationType… in GetFullConnectionPrimitive()181 …attr->activation_type = static_cast<mindspore::schema::ActivationType>(maxPoolAttr.activationType); in GetMaxPoolFusionPrimitive()207 … attr->activation_type = static_cast<mindspore::schema::ActivationType>(matmulAttr.activationType); in GetMatMulFusionPrimitive()277 … attr->activation_type = static_cast<mindspore::schema::ActivationType>(scaleAttr.activationType); in GetScaleFusionPrimitive()304 attr->activation_type = static_cast<mindspore::schema::ActivationType>(actAttr.activationType); in GetActivationPrimitive()352 attr->activation_type = static_cast<mindspore::schema::ActivationType>(mulAttr.activationType); in GetMulFusionPrimitive()
145 auto activation_type = mindspore::lite::ACTIVATION_TYPE_NO_ACTIVATION; in BuildLiteGraph() local146 node->primitive_ = mindspore::lite::MindIR_AddFusion_CreatePrimitive(activation_type); in BuildLiteGraph()