ONE - On-device Neural Engine
Loading...
Searching...
No Matches
StaticMemoryManager.cpp
Go to the documentation of this file.
1
/*
2
* Copyright (c) 2021 Samsung Electronics Co., Ltd. All Rights Reserved
3
*
4
* Licensed under the Apache License, Version 2.0 (the "License");
5
* you may not use this file except in compliance with the License.
6
* You may obtain a copy of the License at
7
*
8
* http://www.apache.org/licenses/LICENSE-2.0
9
*
10
* Unless required by applicable law or agreed to in writing, software
11
* distributed under the License is distributed on an "AS IS" BASIS,
12
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13
* See the License for the specific language governing permissions and
14
* limitations under the License.
15
*/
16
17
#include "
luci_interpreter/StaticMemoryManager.h
"
18
19
namespace
luci_interpreter
20
{
21
22
void
StaticMemoryManager::allocate_memory
(
luci_interpreter::Tensor
&tensor)
23
{
24
if
(!tensor.is_allocatable())
25
{
26
return
;
27
}
28
int32_t
offset
= tensor.get_offset();
29
assert(
offset
>= 0);
30
auto
tensor_ptr = _buffer_ptr +
offset
;
31
tensor.set_data_buffer(tensor_ptr);
32
}
33
34
void
StaticMemoryManager::release_memory
(
luci_interpreter::Tensor
&tensor)
35
{
36
tensor.set_data_buffer(
nullptr
);
37
}
38
39
}
// namespace luci_interpreter
luci_interpreter::StaticMemoryManager::release_memory
void release_memory(luci_interpreter::Tensor &tensor) final
Definition
StaticMemoryManager.cpp:34
luci_interpreter::StaticMemoryManager::allocate_memory
void allocate_memory(luci_interpreter::Tensor &tensor) final
Definition
StaticMemoryManager.cpp:22
luci_interpreter::Tensor
Definition
Tensor.h:101
StaticMemoryManager.h
offset
__global uchar * offset(const Image *img, int x, int y)
Definition
helpers.h:540
luci_interpreter
Definition
BuddyMemoryManager.h:22
compiler
luci-interpreter
src
StaticMemoryManager.cpp
Generated by
1.9.8