Searched refs:depth_attr (Results 1 – 2 of 2) sorted by relevance
133 DenseIntElementsAttr depth_attr; in Verify() local134 if (matchPattern(op.depth(), m_Constant(&depth_attr))) { in Verify()135 if (depth_attr.getType().getRank() != 0) in Verify()137 int64_t depth = depth_attr.getValue<APInt>({}).getSExtValue(); in Verify()160 DenseIntElementsAttr depth_attr; in InferOneHotOpType() local161 if (matchPattern(depth, m_Constant(&depth_attr)) && in InferOneHotOpType()162 depth_attr.getNumElements() == 1) in InferOneHotOpType()163 depth_val = (*depth_attr.begin()).getSExtValue(); in InferOneHotOpType()
4497 DenseIntElementsAttr depth_attr; in matchAndRewrite() local4498 if (!matchPattern(op.depth(), m_Constant(&depth_attr))) { in matchAndRewrite()4502 int64_t depth = depth_attr.getValue<APInt>({}).getSExtValue(); in matchAndRewrite()