Searched refs:tf_squeeze_op (Results 1 – 1 of 1) sorted by relevance
746 auto tf_squeeze_op = cast<TF::SqueezeOp>(op); in matchAndRewrite() local749 auto squeeze_dims_attr = tf_squeeze_op.squeeze_dimsAttr(); in matchAndRewrite()756 convertSqueezeOp(rewriter, op, tf_squeeze_op.getResult(), in matchAndRewrite()757 tf_squeeze_op.input(), squeeze_dims); in matchAndRewrite()