Searched refs:tfl_sum_op (Results 1 – 1 of 1) sorted by relevance
1916 auto tfl_sum_op = cast<TFL::SumOp>(op); in matchAndRewrite() local1919 tfl_sum_op.getResult().getType().dyn_cast<RankedTensorType>(); in matchAndRewrite()1923 if (!matchPattern(tfl_sum_op.axes(), m_Constant(&axes_elems))) in matchAndRewrite()1927 rewriter, op, output_type, tfl_sum_op.input(), axes_elems); in matchAndRewrite()