Searched refs:axisAttrElems (Results 1 – 2 of 2) sorted by relevance
1551 ElementsAttr axisAttrElems; in matchAndRewrite() local1552 if (matchPattern(tf_split_op.split_dim(), m_Constant(&axisAttrElems))) { in matchAndRewrite()1553 axis = axisAttrElems.getValue<IntegerAttr>({}).getInt(); in matchAndRewrite()1585 ElementsAttr axisAttrElems; in matchAndRewrite() local1586 if (!matchPattern(tf_splitv_op.split_dim(), m_Constant(&axisAttrElems))) { in matchAndRewrite()1590 int32_t axis = axisAttrElems.getValue<IntegerAttr>(0).getInt(); in matchAndRewrite()
2062 ElementsAttr axisAttrElems; in matchAndRewrite() local2063 if (!matchPattern(tfl_split_op.split_dim(), m_Constant(&axisAttrElems))) { in matchAndRewrite()2069 int32_t axis = axisAttrElems.getValue<IntegerAttr>({}).getInt(); in matchAndRewrite()2100 ElementsAttr axisAttrElems; in matchAndRewrite() local2101 if (!matchPattern(tfl_splitv_op.split_dim(), m_Constant(&axisAttrElems))) { in matchAndRewrite()2107 int32_t axis = axisAttrElems.getValue<IntegerAttr>(0).getInt(); in matchAndRewrite()