blob: 317838a3140bf482a54374d5fed4696704537718 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
|
/*
* Copyright (c) 2018 Samsung Electronics Co., Ltd. All Rights Reserved
* Copyright (C) 2017 The Android Open Source Project
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#include "ExecutionBuilder.h"
#include "CompilationBuilder.h"
#include "Utils.h"
namespace nnfw {
namespace rt {
CompilationBuilder::CompilationBuilder(const ModelBuilder* model) :
mModel(model) {
VLOG(COMPILATION) << "CompilationBuilder::CompilationBuilder";
}
int CompilationBuilder::finish() {
if (mFinished) {
LOG(ERROR) << "ANeuralNetworksCompilation_finish called more than once";
return ANEURALNETWORKS_BAD_STATE;
}
// TODO validate the rest
mFinished = true;
return ANEURALNETWORKS_NO_ERROR;
}
int CompilationBuilder::setPreference(int32_t preference) {
if (mFinished) {
LOG(ERROR) <<
"ANeuralNetworksCompilation_setPreference can't modify after compilation finished";
return ANEURALNETWORKS_BAD_STATE;
}
if (preference >= kNumberOfPreferences) {
LOG(ERROR) << "ANeuralNetworksCompilation_setPreference invalid preference " << preference;
return ANEURALNETWORKS_BAD_DATA;
}
mPreference = preference;
return ANEURALNETWORKS_NO_ERROR;
}
int CompilationBuilder::createExecution(ExecutionBuilder **execution) {
if (!mFinished) {
LOG(ERROR) << "ANeuralNetworksExecution_create passed an unfinished compilation";
*execution = nullptr;
return ANEURALNETWORKS_BAD_STATE;
}
*execution = new ExecutionBuilder(this);
return (*execution ? ANEURALNETWORKS_NO_ERROR : ANEURALNETWORKS_OUT_OF_MEMORY);
}
} // namespace rt
} // namespace nnfw
|
