Searched refs:stb_bs_attr (Results 1 – 1 of 1) sorted by relevance
312 ElementsAttr stb_bs_attr, bts_bs_attr; in ExtractDilationsAttrFromBlockShape() local313 if (!matchPattern(stb_block_shape, m_Constant(&stb_bs_attr)) || in ExtractDilationsAttrFromBlockShape()319 if (stb_bs_attr.getNumElements() != bts_bs_attr.getNumElements()) return {}; in ExtractDilationsAttrFromBlockShape()320 for (uint64_t i = 0, end = stb_bs_attr.getNumElements(); i < end; ++i) { in ExtractDilationsAttrFromBlockShape()321 if (stb_bs_attr.getValue({i}) != bts_bs_attr.getValue({i})) return {}; in ExtractDilationsAttrFromBlockShape()326 if (stb_bs_attr.getNumElements() >= 2) { in ExtractDilationsAttrFromBlockShape()327 dilation_h_factor = stb_bs_attr.getValue({0}).cast<IntegerAttr>().getInt(); in ExtractDilationsAttrFromBlockShape()328 dilation_w_factor = stb_bs_attr.getValue({1}).cast<IntegerAttr>().getInt(); in ExtractDilationsAttrFromBlockShape()329 } else if (stb_bs_attr.getNumElements() == 1) { in ExtractDilationsAttrFromBlockShape()337 stb_bs_attr.getValue({0}).cast<IntegerAttr>().getInt(); in ExtractDilationsAttrFromBlockShape()[all …]