未验证 提交 e7c7280f 编写于 作者: Z zhangkaihuo 提交者: GitHub

Fix conv api name (#44636)

上级 efb4d5c2
......@@ -80,7 +80,7 @@
data_type : x
backward : cast_grad
- api : conv3d_coo
- api : conv3d
args : (Tensor x, Tensor kernel, int[] paddings, int[] dilations, int[] strides, int groups, bool subm, str key)
output : Tensor(out), Tensor(rulebook), Tensor(counter)
kernel :
......
......@@ -76,7 +76,7 @@ void TestConv3dBase(const std::vector<int>& indices,
kernel.size() * sizeof(T));
if (!std::is_same<T, phi::dtype::float16>::value) {
auto tensor_out = paddle::experimental::sparse::conv3d_coo(
auto tensor_out = paddle::experimental::sparse::conv3d(
x, weight, paddings, dilations, strides, 1, false, "Conv3d");
auto out =
......
......@@ -63,9 +63,9 @@ def _conv3d(x,
dilation = convert_to_list(dilation, dims, 'dilation')
op_type = "conv3d"
pre_bias = _C_ops.final_state_sparse_conv3d_coo(
x, weight, padding, dilation, stride, groups, subm,
key if key is not None else "")
pre_bias = _C_ops.final_state_sparse_conv3d(x, weight, padding, dilation,
stride, groups, subm,
key if key is not None else "")
if bias is not None:
values = pre_bias.values()
add_bias = elementwise_add(values, bias, axis=1)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册