Searched refs:axis_elem (Results 1 – 1 of 1) sorted by relevance
1749 ElementsAttr axis_elem; in matchAndRewrite() local1750 if (!matchPattern(tf_gather_op.axis(), m_Constant(&axis_elem))) in matchAndRewrite()1752 assert(axis_elem.getType().getRank() == 0 && "expected 0D tensor"); in matchAndRewrite()1765 rewriter.getI32IntegerAttr(axis_elem.getValue<IntegerAttr>({}).getInt())); in matchAndRewrite()