已合并
scatter_update和scatter_elements_v2cleancode处理。 #2742
xiaodong666创建于 3月16日
scatter_update和scatter_elements_v2cleancode处理。 #2742
已合并
共 2 个文件变更+6-3
| @@ -1870,8 +1870,11 @@ VC8@ops-nn: | |||
| 1870 | style: | 1870 | style: |
| 1871 | - ops/ops-nn/index/scatter_elements/op_host/arch35/ | 1871 | - ops/ops-nn/index/scatter_elements/op_host/arch35/ |
| 1872 | - ops/ops-nn/index/scatter_elements/op_kernel/arch35/ | 1872 | - ops/ops-nn/index/scatter_elements/op_kernel/arch35/ |
| 1873 | - - ops/ops-nn/index/scatter_elements_v2/op_host/ | 1873 | + - ops/ops-nn/index/scatter_elements_v2/op_host/scatter_elements_v2_asc_tiling.cpp |
| 1874 | - - ops/ops-nn/index/scatter_elements_v2/op_kernel/ | 1874 | + - ops/ops-nn/index/scatter_elements_v2/op_host/scatter_elements_v2_asc_tiling.h |
| 1875 | + - ops/ops-nn/index/scatter_elements_v2/op_kernel/scatter_elements_deterministic.h | ||
| 1876 | + - ops/ops-nn/index/scatter_elements_v2/op_kernel/scatter_elements_v2_apt.cpp | ||
| 1877 | + - ops/ops-nn/index/scatter_elements_v2/op_kernel/scatter_elements.h | ||
| 1875 | - ops/ops-nn/index/masked_scatter/op_host/arch35/ | 1878 | - ops/ops-nn/index/masked_scatter/op_host/arch35/ |
| 1876 | - ops/ops-nn/index/masked_scatter/op_kernel/arch35/ | 1879 | - ops/ops-nn/index/masked_scatter/op_kernel/arch35/ |
| 1877 | - ops/ops-nn/index/scatter_nd_add/op_host/arch35/ | 1880 | - ops/ops-nn/index/scatter_nd_add/op_host/arch35/ |
| @@ -229,7 +229,7 @@ static const aclTensor* TransposeBySpecifiedAxis(const aclTensor* self, int64_t | |||
| 229 | return selfTransposed; | 229 | return selfTransposed; |
| 230 | } | 230 | } |
| 231 | 231 | ||
| 232 | -static aclnnStatus ContiguousAndReshapeParams(aclTensor* selfRef, int64_t& dim, const aclTensor* index, const aclTensor* source, | 232 | +static aclnnStatus ContiguousAndReshapeParams(const aclTensor* selfRef, int64_t& dim, const aclTensor* index, const aclTensor* source, |
| 233 | aclOpExecutor* uniqueExecutor, const aclTensor*& selfContiguous, | 233 | aclOpExecutor* uniqueExecutor, const aclTensor*& selfContiguous, |
| 234 | const aclTensor*& indexContiguous, const aclTensor*& sourceContiguous, | 234 | const aclTensor*& indexContiguous, const aclTensor*& sourceContiguous, |
| 235 | const aclTensor*& selfRefReShape, const aclTensor*& indexReShape, const aclTensor*& sourceReShape, const op::Shape& rowVector) { | 235 | const aclTensor*& selfRefReShape, const aclTensor*& indexReShape, const aclTensor*& sourceReShape, const op::Shape& rowVector) { |