Lines Matching refs:ArrayAttr
174 Value bias, ArrayAttr pad, in buildConvOpWithQuantInfo()
175 ArrayAttr stride, ArrayAttr dilation) { in buildConvOpWithQuantInfo()
197 ArrayAttr outpad, ArrayAttr stride, in buildTransConvOpWithQuantInfo()
198 ArrayAttr outputShape) { in buildTransConvOpWithQuantInfo()
274 ArrayAttr kernel, ArrayAttr stride, in buildAvgPool2dOpWithQuantInfo()
275 ArrayAttr pad) { in buildAvgPool2dOpWithQuantInfo()
330 static void getI64Values(ArrayAttr arrayAttr, SmallVector<int64_t> &values) { in getI64Values()
336 static void getF64Values(ArrayAttr arrayAttr, SmallVector<double> &values) { in getF64Values()
543 ArrayAttr sizes = SliceOpAdaptor(operands, attributes).getSize(); in inferReturnTypeComponents()
575 ArrayAttr multiples = adaptor.getMultiples(); in inferReturnTypeComponents()
611 ArrayAttr newShape = adaptor.getNewShape(); in inferReturnTypeComponents()
995 getI64Values(attributes.get("kernel").cast<ArrayAttr>(), kernel); in NARY_SHAPE_INFER()
996 getI64Values(attributes.get("stride").cast<ArrayAttr>(), stride); in NARY_SHAPE_INFER()
997 getI64Values(attributes.get("pad").cast<ArrayAttr>(), pad); in NARY_SHAPE_INFER()