Searched refs:tf_relu6_op (Results 1 – 1 of 1) sorted by relevance
161 auto tf_relu6_op = cast<TF::Relu6Op>(op); in matchAndRewrite() local164 tf_relu6_op.getResult().getType().dyn_cast<RankedTensorType>(); in matchAndRewrite()170 op, output_type, tf_relu6_op.features(), rewriter.getI64IntegerAttr(0), in matchAndRewrite()174 op, output_type, tf_relu6_op.features(), rewriter.getI64IntegerAttr(6), in matchAndRewrite()