sparse_api.yaml 680 字节
Newer Older
1
- api : conv3d
Z
zhangkaihuo 已提交
2
  args : (Tensor x, Tensor kernel, int[] paddings, int[] dilations, int[] strides, int groups, bool subm)
3 4 5 6 7
  output : Tensor(out@SparseCooTensor), Tensor(rulebook@DenseTensor)
  kernel :
    func : sparse_conv3d
    layout : x

8
- api : to_dense
9 10 11 12
  args : (Tensor x, Backend backend)
  output : Tensor(out@DenseTensor)
  invoke : to_dense_impl(x, backend)

13
- api : to_sparse_coo
14
  args : (Tensor x, Backend backend, int64 sparse_dim)
15 16 17
  output : Tensor(out@SparseCooTensor)
  invoke : to_sparse_coo_impl(x, backend, sparse_dim)

18
- api : to_sparse_csr
19 20 21
  args : (Tensor x, Backend backend)
  output : Tensor(out@SparseCsrTensor)
  invoke : to_sparse_csr_impl(x, backend)