Searched refs:tfl_mean_op (Results 1 – 1 of 1) sorted by relevance
1894 auto tfl_mean_op = cast<TFL::MeanOp>(op); in matchAndRewrite() local1897 tfl_mean_op.getResult().getType().dyn_cast<RankedTensorType>(); in matchAndRewrite()1901 if (!matchPattern(tfl_mean_op.axis(), m_Constant(&axes_elems))) in matchAndRewrite()1905 rewriter, op, output_type, tfl_mean_op.input(), axes_elems); in matchAndRewrite()