Lines Matching refs:sparse_tensor

3 #CSR = #sparse_tensor.encoding<{map = (d0, d1) -> (d0 : dense, d1 : compressed)}>
13 func.func @sparse_metadata_init() -> !sparse_tensor.storage_specifier<#CSR> {
14   %0 = sparse_tensor.storage_specifier.init : !sparse_tensor.storage_specifier<#CSR>
15   return %0 : !sparse_tensor.storage_specifier<#CSR>
23 func.func @sparse_get_md(%arg0: !sparse_tensor.storage_specifier<#CSR>) -> index {
24   %0 = sparse_tensor.storage_specifier.get %arg0 lvl_sz at 0
25        : !sparse_tensor.storage_specifier<#CSR>
35 func.func @sparse_set_md(%arg0: !sparse_tensor.storage_specifier<#CSR>, %arg1: index)
36           -> !sparse_tensor.storage_specifier<#CSR> {
37   %0 = sparse_tensor.storage_specifier.set %arg0 lvl_sz at 0 with %arg1
38        : !sparse_tensor.storage_specifier<#CSR>
39   return %0 : !sparse_tensor.storage_specifier<#CSR>