Lines Matching defs:meta
39 def test_dense_dense_mm(x, y, **meta):
46 def test_torch_matmul(x, y, **meta):
53 def test_bsr_dense_mm(x, y, **meta):
64 def test_bsr_dense_mm_with_meta(x, y, **meta):
67 def test_func(x=x, y=y, meta=meta): argument
73 def test_bsr_scatter_mm2(x, y, **meta):
86 def test_bsr_scatter_mm6(x, y, **meta):
99 def test_bsr_scatter_mm(x, y, **meta):
111 def test_linear(x, y, **meta):
219 meta = torch.sparse._triton_ops.scatter_mm_meta(m, k, n, bm, bk) variable
221 meta = torch.sparse._triton_ops.bsr_dense_mm_meta(m, k, n, bm, bk) variable
355 meta = dict( variable