/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-roccv/checkouts/latest/include/core/tensor_storage.hpp Source File

/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-roccv/checkouts/latest/include/core/tensor_storage.hpp Source File#

3 min read time

Applies to Linux

rocCV: /home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-roccv/checkouts/latest/include/core/tensor_storage.hpp Source File
tensor_storage.hpp
Go to the documentation of this file.
1 /*
2  * Copyright (c) 2025 Advanced Micro Devices, Inc. All rights reserved.
3  * Permission is hereby granted, free of charge, to any person obtaining a copy
4  * of this software and associated documentation files (the "Software"), to deal
5  * in the Software without restriction, including without limitation the rights
6  * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
7  * copies of the Software, and to permit persons to whom the Software is
8  * furnished to do so, subject to the following conditions:
9  *
10  * The above copyright notice and this permission notice shall be included in
11  * all copies or substantial portions of the Software.
12  *
13  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
14  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
15  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
16  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
17  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
18  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
19  * THE SOFTWARE.
20  */
21 
22 #pragma once
23 
24 #include <stdlib.h>
25 
26 #include "core/util_enums.h"
27 
28 namespace roccv {
35  public:
45 
52  explicit TensorStorage(size_t bytes, eDeviceType device);
54 
60  void* data() const;
61 
67  const eDeviceType device() const;
68 
69  private:
70  eDeviceType m_device;
71  eOwnership m_ownership;
72  void* m_data;
73 };
74 
75 } // namespace roccv
Stores the underlying data of a tensor and is responsible for allocation/freeing of tensor memory....
Definition: tensor_storage.hpp:34
TensorStorage(void *data, eDeviceType device, eOwnership ownership=eOwnership::OWNING)
Creates a new TensorStorage object and takes ownership of the data pointer.
void * data() const
Retrieves a raw pointer to the underlying tensor data.
TensorStorage(size_t bytes, eDeviceType device)
Creates a new TensorStorage object and allocates the requested number of bytes.
const eDeviceType device() const
Retrieves the device that the tensor data is allocated on.
Definition: strided_data_wrap.hpp:33
eOwnership
Describes whether a container should own a resource or only be a view for a resource.
Definition: util_enums.h:75
eDeviceType
Describes the device type. Used to determine where Tensor data should be allocated and whether operat...
Definition: util_enums.h:69