Searched refs:fused_activation_fn (Results 1 – 3 of 3) sorted by relevance
601 auto fused_activation_fn = tfl_add_op.fused_activation_functionAttr(); in matchAndRewrite() local603 if (fused_activation_fn) { in matchAndRewrite()605 convertFusedActivation(rewriter, op, output, fused_activation_fn); in matchAndRewrite()698 auto fused_activation_fn = tfl_sub_op.fused_activation_functionAttr(); in matchAndRewrite() local700 if (fused_activation_fn) { in matchAndRewrite()702 convertFusedActivation(rewriter, op, output, fused_activation_fn); in matchAndRewrite()723 auto fused_activation_fn = tfl_mul_op.fused_activation_functionAttr(); in matchAndRewrite() local725 if (fused_activation_fn) { in matchAndRewrite()727 rewriter, op, result.getValue(), fused_activation_fn); in matchAndRewrite()803 auto fused_activation_fn = tfl_div_op.fused_activation_functionAttr(); in matchAndRewrite() local[all …]
2206 StringAttr fused_activation_fn) { in convertFusedActivation() argument2218 if (fused_activation_fn.getValue() == "TANH") { in convertFusedActivation()2230 if (fused_activation_fn.getValue() == "NONE") { in convertFusedActivation()2232 } else if (fused_activation_fn.getValue() == "RELU") { in convertFusedActivation()2240 } else if (fused_activation_fn.getValue() == "RELU6") { in convertFusedActivation()2251 } else if (fused_activation_fn.getValue() == "RELU_N1_TO_1") { in convertFusedActivation()2272 if (fused_activation_fn.getValue() == "NONE") { in convertFusedActivation()2274 } else if (fused_activation_fn.getValue() == "RELU") { in convertFusedActivation()2281 } else if (fused_activation_fn.getValue() == "RELU6") { in convertFusedActivation()2287 } else if (fused_activation_fn.getValue() == "RELU_N1_TO_1") { in convertFusedActivation()[all …]
162 StringAttr fused_activation_fn);