sparse_api.yaml 807 字节
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
  intermediate : rulebook
8
  backward : conv3d_grad
9

10 11 12 13 14 15 16 17
- api : relu
  args : (Tensor x)
  output : Tensor(out@SparseCooTensor)
  kernel :
    func : sparse_relu
    layout : x
  backward : sparse_relu_grad

18
- api : to_dense
19
  args : (Tensor x)
20
  output : Tensor(out@DenseTensor)
21
  invoke : to_dense_impl(x)
22

23
- api : to_sparse_coo
24
  args : (Tensor x, int64_t sparse_dim)
25
  output : Tensor(out@SparseCooTensor)
26
  invoke : to_sparse_coo_impl(x, sparse_dim)
27

28
- api : to_sparse_csr
29
  args : (Tensor x)
30
  output : Tensor(out@SparseCsrTensor)
31
  invoke : to_sparse_csr_impl(x)