Searched refs:ExtractSingleElementAsInteger (Results 1 – 4 of 4) sorted by relevance
45 IntegerAttr ExtractSingleElementAsInteger(ElementsAttr attr);
39 IntegerAttr ExtractSingleElementAsInteger(ElementsAttr attr) { in ExtractSingleElementAsInteger() function
229 IntegerAttr axis_int = ExtractSingleElementAsInteger(axis); in matchAndRewrite()402 if (ExtractSingleElementAsInteger(k).getInt() != 0) return false; in ConvertTFMatrixDiagV2orV3()409 if (ExtractSingleElementAsInteger(num_rows).getInt() != -1) return false; in ConvertTFMatrixDiagV2orV3()416 if (ExtractSingleElementAsInteger(num_cols).getInt() != -1) return false; in ConvertTFMatrixDiagV2orV3()
49 def ExtractSingleElementAsInteger : NativeCodeCall<50 "ExtractSingleElementAsInteger($_self.cast<ElementsAttr>())">;54 …"$_builder.getI32IntegerAttr(ExtractSingleElementAsInteger($_self.cast<ElementsAttr>()).getInt())"…