ncnn

1
// Tencent is pleased to support the open source community by making ncnn available.
2
//
3
// Copyright (C) 2023 THL A29 Limited, a Tencent company. All rights reserved.
4
//
5
// Licensed under the BSD 3-Clause License (the "License"); you may not use this file except
6
// in compliance with the License. You may obtain a copy of the License at
7
//
8
// https://opensource.org/licenses/BSD-3-Clause
9
//
10
// Unless required by applicable law or agreed to in writing, software distributed
11
// under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR
12
// CONDITIONS OF ANY KIND, either express or implied. See the License for the
13
// specific language governing permissions and limitations under the License.
14

15
#include "split_vulkan.h"
16

17
namespace ncnn {
18

19
Split_vulkan::Split_vulkan()
20
{
21
    support_vulkan = true;
22
    support_image_storage = true;
23
}
24

25
int Split_vulkan::forward(const std::vector<VkMat>& bottom_blobs, std::vector<VkMat>& top_blobs, VkCompute& /*cmd*/, const Option& /*opt*/) const
26
{
27
    const VkMat& bottom_blob = bottom_blobs[0];
28
    for (size_t i = 0; i < top_blobs.size(); i++)
29
    {
30
        top_blobs[i] = bottom_blob;
31
    }
32

33
    return 0;
34
}
35

36
int Split_vulkan::forward(const std::vector<VkImageMat>& bottom_blobs, std::vector<VkImageMat>& top_blobs, VkCompute& /*cmd*/, const Option& /*opt*/) const
37
{
38
    const VkImageMat& bottom_blob = bottom_blobs[0];
39
    for (size_t i = 0; i < top_blobs.size(); i++)
40
    {
41
        top_blobs[i] = bottom_blob;
42
    }
43

44
    return 0;
45
}
46

47
} // namespace ncnn
48