From fd39b8714688aa6644c511b6d507b682a994e29a Mon Sep 17 00:00:00 2001 From: "wangwenhua1@huawei.com" <89594672@qq.com> Date: Thu, 12 Nov 2020 17:06:04 +0800 Subject: [PATCH] slice tensor no originshape --- ge/host_kernels/slice_kernel.cc | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/ge/host_kernels/slice_kernel.cc b/ge/host_kernels/slice_kernel.cc index fc98e8a5..e926b930 100644 --- a/ge/host_kernels/slice_kernel.cc +++ b/ge/host_kernels/slice_kernel.cc @@ -99,8 +99,9 @@ Status SliceKernel::Compute(const OpDescPtr attr, const std::vectorGetOutputDesc(0); + GeTensorDesc output_tensor_desc(attr_output_tensor_desc); + output_tensor_desc.SetShape(output_dims); GeTensorPtr output_ptr = MakeShared(output_tensor_desc); if (output_ptr == nullptr) { GELOGW("make_shared ge::GeTensor failed, node name %s.", attr->GetName().c_str());