Lines Matching refs:CSR
6 #CSR = #sparse_tensor.encoding<{
29 // Computes C = A x B with all matrices sparse (SpMSpM) in CSR.
31 func.func @matmul2(%A: tensor<4x8xf64, #CSR>,
32 %B: tensor<8x4xf64, #CSR>) -> tensor<4x4xf64, #CSR> {
33 %C = bufferization.alloc_tensor() : tensor<4x4xf64, #CSR>
35 ins(%A, %B: tensor<4x8xf64, #CSR>, tensor<8x4xf64, #CSR>)
36 outs(%C: tensor<4x4xf64, #CSR>) -> tensor<4x4xf64, #CSR>
37 return %D: tensor<4x4xf64, #CSR>
96 %a1 = sparse_tensor.convert %da : tensor<4x8xf64> to tensor<4x8xf64, #CSR>
98 %a3 = sparse_tensor.convert %sa : tensor<4x8xf64> to tensor<4x8xf64, #CSR>
100 %b1 = sparse_tensor.convert %db : tensor<8x4xf64> to tensor<8x4xf64, #CSR>
102 %b3 = sparse_tensor.convert %sb : tensor<8x4xf64> to tensor<8x4xf64, #CSR>
109 : (tensor<4x8xf64, #CSR>,
110 tensor<8x4xf64, #CSR>) -> tensor<4x4xf64, #CSR>
119 : (tensor<4x8xf64, #CSR>,
120 tensor<8x4xf64, #CSR>) -> tensor<4x4xf64, #CSR>
129 : (tensor<4x8xf64, #CSR>,
130 tensor<8x4xf64, #CSR>) -> tensor<4x4xf64, #CSR>
150 %c1 = sparse_tensor.convert %1 : tensor<4x4xf64, #CSR> to tensor<4x4xf64>
179 %c4 = sparse_tensor.convert %4 : tensor<4x4xf64, #CSR> to tensor<4x4xf64>
202 %c7 = sparse_tensor.convert %7 : tensor<4x4xf64, #CSR> to tensor<4x4xf64>
219 %val7 = sparse_tensor.values %7 : tensor<4x4xf64, #CSR> to memref<?xf64>
227 bufferization.dealloc_tensor %a1 : tensor<4x8xf64, #CSR>
229 bufferization.dealloc_tensor %a3 : tensor<4x8xf64, #CSR>
231 bufferization.dealloc_tensor %b1 : tensor<8x4xf64, #CSR>
233 bufferization.dealloc_tensor %b3 : tensor<8x4xf64, #CSR>
235 bufferization.dealloc_tensor %1 : tensor<4x4xf64, #CSR>
237 bufferization.dealloc_tensor %4 : tensor<4x4xf64, #CSR>
239 bufferization.dealloc_tensor %7 : tensor<4x4xf64, #CSR>