未验证 提交 a1e96e47 编写于 作者: L limingshu 提交者: GitHub

first commit (#50808)

上级 8d325d82
...@@ -36,19 +36,8 @@ void SliceCompute(const Context& ctx, ...@@ -36,19 +36,8 @@ void SliceCompute(const Context& ctx,
// Step 1: Get the accurate attribute value of starts and ends // Step 1: Get the accurate attribute value of starts and ends
std::vector<int64_t> starts = starts_t; std::vector<int64_t> starts = starts_t;
std::vector<int64_t> ends = ends_t; std::vector<int64_t> ends = ends_t;
PADDLE_ENFORCE_EQ(
starts.size(),
axes.size(),
phi::errors::InvalidArgument(
"The size of starts must be equal to the size of axes."));
PADDLE_ENFORCE_EQ(ends.size(),
axes.size(),
phi::errors::InvalidArgument(
"The size of ends must be equal to the size of axes."));
// Step 2: Compute output // Step 2: Compute output
auto in = &input; auto in = &input;
auto in_dims = in->dims(); auto in_dims = in->dims();
auto out_dims = out->dims(); auto out_dims = out->dims();
auto slice_dims = out_dims; auto slice_dims = out_dims;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册