sparse_bw_api.yaml 677 字节
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 5
  output : Tensor(x_grad@DenseTensor), Tensor(kernel_grad@DenseTensor)
  kernel :
Z
zhangkaihuo 已提交
6
    func : sparse_conv3d_grad
Z
zhangkaihuo 已提交
7 8 9 10 11 12 13

- backward_api : sparse_relu_grad
  forward : sparse_relu(Tensor x) -> Tensor(out@SparseCooTensor)
  args : (Tensor x, Tensor out_grad)
  output : Tensor(x_grad@SparseCooTensor)
  kernel : 
    func : sparse_relu_grad