From 001982ce1419504f1f0e1d116d57dd34f0180008 Mon Sep 17 00:00:00 2001 From: Masahiro Masuda Date: Mon, 24 May 2021 17:26:56 +0900 Subject: [PATCH] format --- include/tvm/topi/transform.h | 28 +++++++++++++--------------- 1 file changed, 13 insertions(+), 15 deletions(-) diff --git a/include/tvm/topi/transform.h b/include/tvm/topi/transform.h index 921f8013586f..edb6bbed01e8 100644 --- a/include/tvm/topi/transform.h +++ b/include/tvm/topi/transform.h @@ -687,7 +687,7 @@ inline Array StridedSliceCanonicalizeBegin(const Array& isha } else { auto idim = ishape[axes[i]]; auto b_expr = make_const(dtype, begin[i]); - PrimExpr b = begin[i] < 0 ? b_expr + idim : b_expr; + PrimExpr b = begin[i] < 0 ? b_expr + idim : b_expr; auto s = strides[i]; if (s < 0) { b = tvm::min(b, idim - 1); @@ -700,12 +700,13 @@ inline Array StridedSliceCanonicalizeBegin(const Array& isha return begin_expr; } -inline Array StridedSliceOutputShape(const Array& ishape, const std::vector& begin, +inline Array StridedSliceOutputShape(const Array& ishape, + const std::vector& begin, const std::vector& end, const std::vector& strides, const Array& axes, std::string slice_mode, const Array& begin_canonicalized, - bool use_any=false) { + bool use_any = false) { size_t src_tensor_dim = ishape.size(); Array out_shape; for (size_t i = 0; i < src_tensor_dim; ++i) { @@ -734,21 +735,18 @@ inline Array StridedSliceOutputShape(const Array& ishape, co return out_shape; } -inline Array StridedSliceOutputShape(const Array& ishape, const Array& begin, - const Array& end, - const Array& strides, - const Array& axes, - const std::string& slice_mode) { +inline Array StridedSliceOutputShape( + const Array& ishape, const Array& begin, const Array& end, + const Array& strides, const Array& axes, const std::string& slice_mode) { ICHECK(axes.size() == begin.size() && axes.size() == end.size() && axes.size() == strides.size()); std::vector begin_vec, end_vec, strides_vec; std::tie(begin_vec, end_vec, strides_vec) = ToVec(begin, end, strides, slice_mode); - auto begin_canonicalized = - StridedSliceCanonicalizeBegin(ishape, begin_vec, strides_vec, axes, begin[0]->dtype, slice_mode); + auto begin_canonicalized = StridedSliceCanonicalizeBegin(ishape, begin_vec, strides_vec, axes, + begin[0]->dtype, slice_mode); return StridedSliceOutputShape(ishape, begin_vec, end_vec, strides_vec, axes, slice_mode, begin_canonicalized, true); } - /*! * \brief strided_slice of a tensor * @@ -775,10 +773,10 @@ inline Tensor strided_slice_with_axes(const Tensor& x, const Array& beg std::vector begin_vec, end_vec, strides_vec; std::tie(begin_vec, end_vec, strides_vec) = ToVec(begin, end, strides, slice_mode); - auto begin_expr = - StridedSliceCanonicalizeBegin(x->shape, begin_vec, strides_vec, axes, begin[0]->dtype, slice_mode); - auto out_shape = - StridedSliceOutputShape(x->shape, begin_vec, end_vec, strides_vec, axes, slice_mode, begin_expr); + auto begin_expr = StridedSliceCanonicalizeBegin(x->shape, begin_vec, strides_vec, axes, + begin[0]->dtype, slice_mode); + auto out_shape = StridedSliceOutputShape(x->shape, begin_vec, end_vec, strides_vec, axes, + slice_mode, begin_expr); return te::compute( out_shape,