sparse_bw_api.yaml 458 字节
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