sparse_api.yaml 655 字节
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
- api : to_dense
11
  args : (Tensor x)
12
  output : Tensor(out@DenseTensor)
13
  invoke : to_dense_impl(x)
14

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

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