Searched refs:activation_attr (Results 1 – 2 of 2) sorted by relevance
981 auto& activation_attr = activation->attr(); in IsCpuCompatibleDataType() local982 (*attr)["leakyrelu_alpha"] = activation_attr.at("alpha"); in IsCpuCompatibleDataType()1002 auto& activation_attr = activation->attr(); in IsCpuCompatibleDataType() local1003 (*attr)["leakyrelu_alpha"] = activation_attr.at("alpha"); in IsCpuCompatibleDataType()1045 auto& activation_attr = activation->attr(); in IsCpuCompatibleDataType() local1046 (*attr)["leakyrelu_alpha"] = activation_attr.at("alpha"); in IsCpuCompatibleDataType()
797 StringAttr activation_attr = builder_.getStringAttr( in EmitDnnConvolution() local799 op.activation_modeAttr(activation_attr); in EmitDnnConvolution()