File tree Expand file tree Collapse file tree 3 files changed +32
-2
lines changed Expand file tree Collapse file tree 3 files changed +32
-2
lines changed Original file line number Diff line number Diff line change 1212// See the License for the specific language governing permissions and
1313// limitations under the License.
1414
15+ #include " paddle/phi/kernels/sequence_expand_kernel.h"
1516#include " paddle/phi/core/kernel_registry.h"
1617#include " paddle/phi/kernels/impl/sequence_expand_kernel_impl.h"
17-
1818namespace phi {
1919
2020template <typename T>
Original file line number Diff line number Diff line change 1212// See the License for the specific language governing permissions and
1313// limitations under the License.
1414
15+ #include " paddle/phi/kernels/sequence_expand_kernel.h"
1516#include " paddle/phi/backends/gpu/gpu_primitives.h"
1617#include " paddle/phi/kernels/impl/sequence_expand_kernel_impl.h"
17-
1818namespace phi {
1919
2020template <typename T>
Original file line number Diff line number Diff line change 1+ // Copyright (c) 2024 PaddlePaddle Authors. All Rights Reserved.
2+ //
3+ // Licensed under the Apache License, Version 2.0 (the "License");
4+ // you may not use this file except in compliance with the License.
5+ // You may obtain a copy of the License at
6+ //
7+ // http://www.apache.org/licenses/LICENSE-2.0
8+ //
9+ // Unless required by applicable law or agreed to in writing, software
10+ // distributed under the License is distributed on an "AS IS" BASIS,
11+ // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12+ // See the License for the specific language governing permissions and
13+ // limitations under the License.
14+
15+ #pragma once
16+
17+ #include " paddle/phi/core/dense_tensor.h"
18+ #include " paddle/phi/core/device_context.h"
19+ #include " paddle/phi/core/kernel_registry.h"
20+
21+ namespace phi {
22+
23+ template <typename T, typename Context>
24+ void SequenceExpandKernel (const Context& dev_ctx,
25+ const DenseTensor& x,
26+ const DenseTensor& y,
27+ int ref_level,
28+ DenseTensor* out);
29+
30+ } // namespace phi
You can’t perform that action at this time.
0 commit comments