Browse Source

slice tensor no originshape

tags/v1.1.0
wangwenhua1@huawei.com 4 years ago
parent
commit
e34b9ce6c0
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      ge/host_kernels/slice_kernel.cc

+ 1
- 1
ge/host_kernels/slice_kernel.cc View File

@@ -99,7 +99,7 @@ Status SliceKernel::Compute(const OpDescPtr attr, const std::vector<ConstGeTenso
stride_vec.push_back(1);
}
// construct tensorDesc
atuo attr_output_tensor_desc = attr->GetOutputDesc(0);
auto attr_output_tensor_desc = attr->GetOutputDesc(0);
GeTensorDesc output_tensor_desc(attr_output_tensor_desc);
output_tensor_desc.SetShape(output_dims);
GeTensorPtr output_ptr = MakeShared<GeTensor>(output_tensor_desc);


Loading…
Cancel
Save