Searched refs:tf_sum_op (Results 1 – 1 of 1) sorted by relevance
1188 auto tf_sum_op = cast<TF::SumOp>(op); in matchAndRewrite() local1191 tf_sum_op.getResult().getType().dyn_cast<RankedTensorType>(); in matchAndRewrite()1195 if (!matchPattern(tf_sum_op.reduction_indices(), m_Constant(&axes_elems))) in matchAndRewrite()1199 auto keep_dims_attr = tf_sum_op.keep_dimsAttr(); in matchAndRewrite()1203 rewriter, op, output_type, tf_sum_op.input(), axes_elems, keep_dims); in matchAndRewrite()