Searched refs:axis_elems (Results 1 – 2 of 2) sorted by relevance
478 ElementsAttr axis_elems; in matchAndRewrite() local479 if (!matchPattern(tf_argmax_op.dimension(), m_Constant(&axis_elems))) in matchAndRewrite()482 int32_t axis = axis_elems.getValues<IntegerAttr>()[0].getInt(); in matchAndRewrite()643 ElementsAttr axis_elems; in matchAndRewrite() local644 if (!matchPattern(tf_concatv2_op.axis(), m_Constant(&axis_elems))) in matchAndRewrite()647 int32_t axis = axis_elems.getValues<IntegerAttr>()[0].getInt(); in matchAndRewrite()2046 ElementsAttr axis_elems; in matchAndRewrite() local2047 if (!matchPattern(tf_reverse_op.axis(), m_Constant(&axis_elems))) in matchAndRewrite()2052 if (axis_elems.getNumElements() == 0) { in matchAndRewrite()2057 for (int i = 0; i < axis_elems.getNumElements(); i++) { in matchAndRewrite()[all …]
2948 ElementsAttr axis_elems; in matchAndRewrite() local2949 if (!matchPattern(tfl_reverse_op.axis(), m_Constant(&axis_elems))) in matchAndRewrite()2954 if (axis_elems.getNumElements() == 0) { in matchAndRewrite()2959 for (int i = 0; i < axis_elems.getNumElements(); i++) { in matchAndRewrite()2960 int64_t axis_val = axis_elems.getValues<APInt>()[i].getSExtValue(); in matchAndRewrite()