Compute Library
 20.08
NETensorHandle.cpp
Go to the documentation of this file.
1 /*
2  * Copyright (c) 2018-2019 Arm Limited.
3  *
4  * SPDX-License-Identifier: MIT
5  *
6  * Permission is hereby granted, free of charge, to any person obtaining a copy
7  * of this software and associated documentation files (the "Software"), to
8  * deal in the Software without restriction, including without limitation the
9  * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
10  * sell copies of the Software, and to permit persons to whom the Software is
11  * furnished to do so, subject to the following conditions:
12  *
13  * The above copyright notice and this permission notice shall be included in all
14  * copies or substantial portions of the Software.
15  *
16  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
19  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
21  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
22  * SOFTWARE.
23  */
25 
28 
29 namespace arm_compute
30 {
31 namespace graph
32 {
33 namespace backends
34 {
36  : _tensor()
37 {
38  _tensor.allocator()->init(info);
39 }
40 
42 {
43  _tensor.allocator()->allocate();
44 }
45 
47 {
48  _tensor.allocator()->free();
49 }
50 
52 {
53  if(mg != nullptr)
54  {
55  mg->manage(&_tensor);
56  }
57 }
58 
59 void NETensorHandle::map(bool blocking)
60 {
61  ARM_COMPUTE_UNUSED(blocking);
62 }
63 
65 {
66 }
67 
69 {
70  // TODO (geopin01): Release tensor only if all sub-tensors are marked as not used
71  if(!_tensor.is_used())
72  {
73  _tensor.allocator()->free();
74  }
75 }
76 
78 {
79  return _tensor;
80 }
81 
83 {
84  return _tensor;
85 }
86 
88 {
89  return this;
90 }
91 
93 {
94  return false;
95 }
96 
98 {
99  return Target::NEON;
100 }
101 } // namespace backends
102 } // namespace graph
103 } // namespace arm_compute
NETensorHandle(const ITensorInfo &info)
Default Constructor.
Memory group interface.
Definition: IMemoryGroup.h:37
ITensorHandle * parent_handle() override
Return the parent tensor handle if is a subtensor else this.
NEON capable target device.
void map(bool blocking) override
Maps backend tensor object.
void unmap() override
Un-maps a backend tensor object.
void init(const TensorAllocator &allocator, const Coordinates &coords, TensorInfo &sub_info)
Shares the same backing memory with another tensor allocator, while the tensor info might be differen...
void manage(IMemoryGroup *mg) override
Set backend tensor to be managed by a memory group.
bool is_used() const
Flags if the tensor is used or not.
Definition: ITensor.cpp:163
Store the tensor's metadata.
Definition: ITensorInfo.h:40
Target target() const override
Returns target type.
Interface for NEON tensor.
Definition: ITensor.h:36
Copyright (c) 2017-2020 Arm Limited.
TensorAllocator * allocator()
Return a pointer to the tensor's allocator.
Definition: Tensor.cpp:48
void free() override
Allocates backend memory for the handle.
bool is_subtensor() const override
Checks if a backing tensor is a sub-tensor object or not.
#define ARM_COMPUTE_UNUSED(...)
To avoid unused variables warnings.
Definition: Error.h:152
void allocate() override
Allocate size specified by TensorInfo of CPU memory.
void free() override
Free allocated CPU memory.
Tensor handle interface object.
Definition: ITensorHandle.h:38
arm_compute::ITensor & tensor() override
Backend tensor object accessor.
void allocate() override
Allocates backend memory for the handle.
void release_if_unused() override
Releases backend tensor if is marked as unused.
virtual void manage(IMemoryManageable *obj)=0
Sets a object to be managed by the given memory group.