sparse_api.yaml 627 字节
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
  output : Tensor(out@SparseCooTensor), Tensor(rulebook@DenseTensor)
  kernel :
    func : sparse_conv3d
    layout : x
7
  backward : conv3d_grad
8

9
- api : to_dense
10
  args : (Tensor x)
11
  output : Tensor(out@DenseTensor)
12
  invoke : to_dense_impl(x)
13

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

19
- api : to_sparse_csr
20
  args : (Tensor x)
21
  output : Tensor(out@SparseCsrTensor)
22
  invoke : to_sparse_csr_impl(x)