From e7c7280fff53d3b951d62cfb2a92d52350709b08 Mon Sep 17 00:00:00 2001 From: zhangkaihuo Date: Wed, 27 Jul 2022 12:40:36 +0800 Subject: [PATCH] Fix conv api name (#44636) --- paddle/phi/api/yaml/sparse_api.yaml | 2 +- paddle/phi/tests/api/test_sparse_conv_api.cc | 2 +- python/paddle/incubate/sparse/nn/functional/conv.py | 6 +++--- 3 files changed, 5 insertions(+), 5 deletions(-) diff --git a/paddle/phi/api/yaml/sparse_api.yaml b/paddle/phi/api/yaml/sparse_api.yaml index 904b1854489..e11306f21f2 100644 --- a/paddle/phi/api/yaml/sparse_api.yaml +++ b/paddle/phi/api/yaml/sparse_api.yaml @@ -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 : diff --git a/paddle/phi/tests/api/test_sparse_conv_api.cc b/paddle/phi/tests/api/test_sparse_conv_api.cc index b1df197f42f..123f824f62d 100644 --- a/paddle/phi/tests/api/test_sparse_conv_api.cc +++ b/paddle/phi/tests/api/test_sparse_conv_api.cc @@ -76,7 +76,7 @@ void TestConv3dBase(const std::vector& indices, kernel.size() * sizeof(T)); if (!std::is_same::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 = diff --git a/python/paddle/incubate/sparse/nn/functional/conv.py b/python/paddle/incubate/sparse/nn/functional/conv.py index 60cbb94bea2..605cadc2b09 100644 --- a/python/paddle/incubate/sparse/nn/functional/conv.py +++ b/python/paddle/incubate/sparse/nn/functional/conv.py @@ -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) -- GitLab