Searched refs:tf_relu_op (Results 1 – 1 of 1) sorted by relevance
139 auto tf_relu_op = cast<TF::ReluOp>(op); in matchAndRewrite() local142 tf_relu_op.getResult().getType().dyn_cast<RankedTensorType>(); in matchAndRewrite()148 op, output_type, tf_relu_op.features(), rewriter.getI64IntegerAttr(0), in matchAndRewrite()152 op, output_type, tf_relu_op.features(), in matchAndRewrite()