ONE - On-device Neural Engine
Loading...
Searching...
No Matches
SimpleMemoryManager.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
#ifndef USE_STATIC_ALLOC
18
19
#include "
SimpleMemoryManager.h
"
20
21
namespace
luci_interpreter
22
{
23
24
uint8_t *
SimpleMemoryManager::allocate_memory
(
const
circle::Tensor *tensor)
25
{
26
const
auto
element_size =
getDataTypeSize
(Tensor::element_type(tensor));
27
const
auto
num_elements = Tensor::num_elements(tensor);
28
29
assert(element_size * num_elements > 0);
30
31
return
new
uint8_t[num_elements * element_size];
32
}
33
34
void
SimpleMemoryManager::release_memory
(uint8_t *
data
)
35
{
36
if
(
data
==
nullptr
)
37
return
;
38
39
delete
[]
data
;
40
}
41
42
}
// namespace luci_interpreter
43
44
#endif
// USE_STATIC_ALLOC
luci_interpreter::SimpleMemoryManager::release_memory
void release_memory(luci_interpreter::Tensor &tensor) final
Definition
SimpleMemoryManager.cpp:41
luci_interpreter::SimpleMemoryManager::allocate_memory
void allocate_memory(luci_interpreter::Tensor &tensor) final
Definition
SimpleMemoryManager.cpp:22
flatbuffers::data
const T * data(const std::vector< T, Alloc > &v)
Definition
flatbuffers.h:1145
luci_interpreter
Definition
BuddyMemoryManager.h:22
luci_interpreter::getDataTypeSize
size_t getDataTypeSize(DataType data_type)
Definition
DataType.h:33
SimpleMemoryManager.h
onert-micro
luci-interpreter
src
memory_managers
SimpleMemoryManager.cpp
Generated by
1.9.8