Searched refs:value_elem (Results 1 – 2 of 2) sorted by relevance
790 ElementsAttr value_elem; in matchAndRewrite() local791 if (!matchPattern(tf_fill_op.value(), m_Constant(&value_elem))) in matchAndRewrite()795 ArrayRef<int64_t>(dims_vals), value_elem.getType().getElementType()); in matchAndRewrite()799 if (value_elem.getType().getElementType().isa<FloatType>()) { in matchAndRewrite()802 value_elem.getValues<FloatAttr>()[0].getValue().convertToFloat()); in matchAndRewrite()807 value_elem.getValues<IntegerAttr>()[0].getValue().getLimitedValue()); in matchAndRewrite()
1779 ElementsAttr value_elem; in matchAndRewrite() local1780 if (!matchPattern(tfl_fill_op.input(), m_Constant(&value_elem))) in matchAndRewrite()1784 ArrayRef<int64_t>(dims_vals), value_elem.getType().getElementType()); in matchAndRewrite()1788 if (value_elem.getType().getElementType().isa<FloatType>()) { in matchAndRewrite()1790 total_size, value_elem.getValues<APFloat>()[0].convertToFloat()); in matchAndRewrite()1794 total_size, value_elem.getValues<APInt>()[0].getLimitedValue()); in matchAndRewrite()