Searched refs:axis_elems (Results 1 – 2 of 2) sorted by relevance
473 ElementsAttr axis_elems; in matchAndRewrite() local474 if (!matchPattern(tf_argmax_op.dimension(), m_Constant(&axis_elems))) in matchAndRewrite()477 int32_t axis = axis_elems.getValue<IntegerAttr>({}).getInt(); in matchAndRewrite()638 ElementsAttr axis_elems; in matchAndRewrite() local639 if (!matchPattern(tf_concatv2_op.axis(), m_Constant(&axis_elems))) in matchAndRewrite()642 int32_t axis = axis_elems.getValue<IntegerAttr>({}).getInt(); in matchAndRewrite()1947 ElementsAttr axis_elems; in matchAndRewrite() local1948 if (!matchPattern(tf_reverse_op.axis(), m_Constant(&axis_elems))) in matchAndRewrite()1953 if (axis_elems.getNumElements() == 0) { in matchAndRewrite()1958 for (int i = 0; i < axis_elems.getNumElements(); i++) { in matchAndRewrite()[all …]
2761 ElementsAttr axis_elems; in matchAndRewrite() local2762 if (!matchPattern(tfl_reverse_op.axis(), m_Constant(&axis_elems))) in matchAndRewrite()2767 if (axis_elems.getNumElements() == 0) { in matchAndRewrite()2772 for (int i = 0; i < axis_elems.getNumElements(); i++) { in matchAndRewrite()2773 int64_t axis_val = axis_elems.getValue<IntegerAttr>(i).getInt(); in matchAndRewrite()