sparse_bw_api.yaml 680 字节
Newer Older
1
- backward_api : conv3d_grad
Z
zhangkaihuo 已提交
2 3
  forward : conv3d (Tensor x, Tensor kernel, int[] paddings, int[] dilations, int[] strides, int groups, bool subm) -> Tensor(out@SparseCooTensor), Tensor(rulebook@DenseTensor)
  args : (Tensor x, Tensor kernel, Tensor rulebook, Tensor out_grad, int[] paddings, int[] dilations, int[] strides, int groups, bool subm)
4
  output : Tensor(x_grad@SparseCooTensor), Tensor(kernel_grad@DenseTensor)
5
  kernel :
Z
zhangkaihuo 已提交
6
    func : sparse_conv3d_grad
Z
zhangkaihuo 已提交
7 8 9 10 11

- backward_api : sparse_relu_grad
  forward : sparse_relu(Tensor x) -> Tensor(out@SparseCooTensor)
  args : (Tensor x, Tensor out_grad)
  output : Tensor(x_grad@SparseCooTensor)
Z
zyfncg 已提交
12
  kernel :
Z
zhangkaihuo 已提交
13
    func : sparse_relu_grad