Browse Source

Fix code check

tags/v1.2.0
dongduo 3 years ago
parent
commit
9894748d1d
1 changed files with 4 additions and 0 deletions
  1. +4
    -0
      ge/host_kernels/strided_slice_kernel.cc

+ 4
- 0
ge/host_kernels/strided_slice_kernel.cc View File

@@ -272,6 +272,10 @@ Status StridedSliceKernel::InitParamWithAttrs(const std::vector<ConstGeTensorPtr
void StridedSliceKernel::ExpandDimsWithNewAxis(const ConstGeTensorPtr &begin_tensor, const size_t x_dims_num, void StridedSliceKernel::ExpandDimsWithNewAxis(const ConstGeTensorPtr &begin_tensor, const size_t x_dims_num,
vector<int64_t> &x_dims) { vector<int64_t> &x_dims) {
auto begin_data_type_size = GetSizeByDataType(begin_tensor->GetTensorDesc().GetDataType()); auto begin_data_type_size = GetSizeByDataType(begin_tensor->GetTensorDesc().GetDataType());
if (begin_data_type_size == 0) {
GELOGW("Param begin_data_type_size should not be zero.");
return;
}
size_t begin_vec_size = begin_tensor->GetData().size() / begin_data_type_size; size_t begin_vec_size = begin_tensor->GetData().size() / begin_data_type_size;
auto final_dim_num = x_dims_num < begin_vec_size ? begin_vec_size : x_dims_num; auto final_dim_num = x_dims_num < begin_vec_size ? begin_vec_size : x_dims_num;
for (size_t i = 0; i < final_dim_num; i++) { for (size_t i = 0; i < final_dim_num; i++) {


Loading…
Cancel
Save