Lines Matching refs:sparse_tensor

3 #COO = #sparse_tensor.encoding<{
14 // CHECK:           %[[VAL_3:.*]] = sparse_tensor.extract_iteration_space %[[VAL_0]] lvls = 0 to 2
15 // CHECK:           %[[VAL_4:.*]] = sparse_tensor.iterate %[[VAL_5:.*]] in %[[VAL_3]] iter_args(%[[…
17 // CHECK:             sparse_tensor.yield %[[VAL_7]] : index
24   %l1 = sparse_tensor.extract_iteration_space %sp lvls = 0
25       : tensor<4x8xf32, #COO> -> !sparse_tensor.iter_space<#COO, lvls = 0>
26 …%r1 = sparse_tensor.iterate %it1 in %l1 iter_args(%outer = %i): !sparse_tensor.iter_space<#COO, lv…
27     %l2 = sparse_tensor.extract_iteration_space %sp at %it1 lvls = 1
28 …: tensor<4x8xf32, #COO>, !sparse_tensor.iterator<#COO, lvls = 0 to 1> -> !sparse_tensor.iter_space…
29 …%r2 = sparse_tensor.iterate %it2 in %l2 iter_args(%inner = %outer): !sparse_tensor.iter_space<#COO…
31       sparse_tensor.yield %k : index
33     sparse_tensor.yield %r2 : index