Lines Matching refs:strides

10 func.func @dot(%x: memref<?xf32, offset: ?, strides: [1]>,
11           %y: memref<?xf32, offset: ?, strides: [1]>,
14     ins(%x, %y: memref<?xf32, offset: ?, strides: [1]>,
15                 memref<?xf32, offset: ?, strides: [1]>)
33 func.func @matvec(%A: memref<?x?xf32, offset: ?, strides: [?, 1]>,
34              %x: memref<?xf32, offset: ?, strides: [1]>,
35              %y: memref<?xf32, offset: ?, strides: [1]>) {
37     ins(%A, %x: memref<?x?xf32, offset: ?, strides: [?, 1]>,
38                 memref<?xf32, offset: ?, strides: [1]>)
39     outs(%y: memref<?xf32, offset: ?, strides: [1]>)
52 func.func @matmul(%A: memref<?x?xf32, offset: ?, strides: [?, 1]>,
53              %B: memref<?x?xf32, offset: ?, strides: [?, 1]>,
54              %C: memref<?x?xf32, offset: ?, strides: [?, 1]>) {
56     ins(%A, %B: memref<?x?xf32, offset: ?, strides: [?, 1]>,
57                 memref<?x?xf32, offset: ?, strides: [?, 1]>)
58     outs(%C: memref<?x?xf32, offset: ?, strides: [?, 1]>)
103 func.func @permute_generic(%A: memref<?x?xf32, offset: ?, strides: [?, 1]>,
104            %B: memref<?x?xf32, offset: ?, strides: [?, 1]>,
105            %C: memref<?x?xf32, offset: ?, strides: [?, 1]>) {
107     ins(%A, %B : memref<?x?xf32, offset: ?, strides: [?, 1]>,
108                  memref<?x?xf32, offset: ?, strides: [?, 1]>)
109    outs(%C : memref<?x?xf32, offset: ?, strides: [?, 1]>) {
126 func.func @matvec_perm(%A: memref<?x?xf32, offset: ?, strides: [?, 1]>,
127              %x: memref<?xf32, offset: ?, strides: [1]>,
128              %y: memref<?xf32, offset: ?, strides: [1]>) {
130     ins(%A, %x: memref<?x?xf32, offset: ?, strides: [?, 1]>,
131                 memref<?xf32, offset: ?, strides: [1]>)
132    outs(%y: memref<?xf32, offset: ?, strides: [1]>)
145 func.func @matmul_perm(%A: memref<?x?xf32, offset: ?, strides: [?, 1]>,
146              %B: memref<?x?xf32, offset: ?, strides: [?, 1]>,
147              %C: memref<?x?xf32, offset: ?, strides: [?, 1]>) {
149     ins(%A, %B: memref<?x?xf32, offset: ?, strides: [?, 1]>,
150                 memref<?x?xf32, offset: ?, strides: [?, 1]>)
151    outs(%C : memref<?x?xf32, offset: ?, strides: [?, 1]>)