diff --git a/include/tvm/topi/transform.h b/include/tvm/topi/transform.h index 36e8dd5958c7..598c862ababc 100644 --- a/include/tvm/topi/transform.h +++ b/include/tvm/topi/transform.h @@ -758,10 +758,10 @@ inline Tensor strided_slice(const Tensor& x, const Array& begin, const Array strides_full(strides); const IntImm one = IntImm(DataType::Int(64), 1); - const IntImm zero = IntImm(DataType::Int(64), 1); + const IntImm zero = IntImm(DataType::Int(64), 0); const IntImm max_range = IntImm(DataType::Int(64), std::numeric_limits::max()); - for (size_t i = strides_full.size(); i < src_tensor_dim; ++i) { + for (size_t i = strides.size(); i < src_tensor_dim; ++i) { strides_full.push_back(one); } for (size_t i = begin.size(); i < src_tensor_dim; ++i) {