Searched refs:activation_mode_str (Results 1 – 1 of 1) sorted by relevance
68 string activation_mode_str; in ParseActivationMode() local69 TF_RETURN_IF_ERROR(context->GetAttr("activation_mode", &activation_mode_str)); in ParseActivationMode()71 if (activation_mode_str == "Identity") { in ParseActivationMode()75 if (activation_mode_str == "Relu") { in ParseActivationMode()80 activation_mode_str); in ParseActivationMode()